Advertisement
clown1337

Untitled

May 20th, 2024
384
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.94 KB | None | 0 0
  1. #ifdef ONPC
  2. #define _GLIBCXX_DEBUG
  3. #endif
  4. #include <bits/stdc++.h>
  5.  
  6. using namespace std;
  7. /// Pragmas ///
  8. /// define ///
  9. #define pii pair<int, int>
  10. #define pll pair<ll, ll>
  11. #define V vector
  12. #define MP make_pair
  13. #define vi vector<int>
  14. #define ff first
  15. #define ss second
  16. #define vl vector<long long>
  17. /// typedef ///
  18. typedef long long ll;
  19. typedef long double ld;
  20. typedef unsigned long long ull;
  21.  
  22. /// solve ///
  23.  
  24. const int MOD = 1000000007;
  25. int n = 3;
  26.  
  27. int get(int mask, int i) { return (mask >> i) & 1; }
  28.  
  29. bool check(int mask1, int mask2) {
  30.   int cnt = 0;
  31.   for (int i = 0; i < n; i++) {
  32.     if (get(mask1, i) == 0 && get(mask2, i) != 1) {
  33.       return false;
  34.     } else {
  35.       if (get(mask1, i) == 1 && get(mask2, i) == 1) {
  36.         cnt++;
  37.       } else {
  38.         if (cnt % 2 != 0) {
  39.           return false;
  40.         } else {
  41.           cnt = 0;
  42.         }
  43.       }
  44.     }
  45.   }
  46.   if (cnt % 2 != 0) {
  47.     return false;
  48.   } else {
  49.     return true;
  50.   }
  51. }
  52. void solve() {
  53.   int m;
  54.   cin >> m;
  55.   vector<vector<int>> dp(m + 1, vector<int>((1 << n), 0));
  56.     dp[0][(1 << n) - 1] = 1;
  57.   for (int i = 0; i < m; i++) {
  58.     for (int p = 0; p < (1 << n); p++) {
  59.       for (int q = 0; q < (1 << n); q++) {
  60.         if (check(p, q)) {
  61.           dp[i+1][q] += dp[i][p];
  62.                     dp[i][q] %= MOD;
  63.         }
  64.       }
  65.     }
  66.   }
  67.   cout << dp[m][(1 << n) - 1] % MOD << '\n';
  68. }
  69.  
  70. signed main() {
  71.   ios_base::sync_with_stdio(false);
  72.   cin.tie(NULL);
  73.   cout.tie(NULL);
  74.   freopen("input.txt", "r", stdin);
  75.   freopen("output.txt", "w", stdout);
  76.   int t = 1;
  77.   // cin >> t;
  78. #ifdef ONPC
  79.   std::cout << std::unitbuf;
  80. #endif
  81.   while (t--)
  82.     solve();
  83. #ifdef ONPC
  84.   cout << endl << "__________________________" << endl;
  85. #endif
  86. #ifdef ONPC
  87.   cout << endl
  88.        << "finished in " << clock() * 1.0 / CLOCKS_PER_SEC << " sec" << endl;
  89.   cerr << endl
  90.        << "finished in " << clock() * 1.0 / CLOCKS_PER_SEC << " sec" << endl;
  91. #endif
  92.   return 0;
  93. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement