Guest User

Untitled

a guest
Apr 30th, 2020
547
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.28 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define int long long
  5. #define f(i,n) for(int i=0;i<n;i++)
  6. #define fast ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0)
  7.  
  8. const int MOD = (1LL<<32)-1;
  9. const int SZ = 50;
  10.  
  11. struct matrix
  12. {
  13.     int a[SZ][SZ];
  14.  
  15.     matrix()
  16.     {
  17.         memset(a,0,sizeof(a));
  18.     }
  19.  
  20.     void iden()
  21.     {
  22.         f(i,SZ) a[i][i] = 1;
  23.     }
  24.  
  25.     matrix operator*(matrix & o)
  26.     {
  27.         matrix res;
  28.  
  29.         f(i,SZ) f(j,SZ) f(k,SZ)
  30.             res.a[i][j]+=(a[i][k]*o.a[k][j])&MOD;
  31.  
  32.         f(i,SZ) f(j,SZ) res.a[i][j]&=MOD;
  33.  
  34.         return res;
  35.     }
  36.  
  37. };
  38.  
  39. matrix pow(matrix a,int b)
  40. {
  41.    matrix res;
  42.    res.iden();
  43.  
  44.     while(b > 0)
  45.     {
  46.         if(b&1) res = (res*a);
  47.         a = (a*a);
  48.         b>>=1;
  49.     }
  50.  
  51.     return res;
  52. }
  53.  
  54. vector<int> prefix_function(string s)
  55. {
  56.     int n = (int)s.length();
  57.     vector<int> pi(n);
  58.  
  59.     for (int i = 1; i < n; i++)
  60.     {
  61.         int j = pi[i-1];
  62.         while (j > 0 && s[i] != s[j])  j = pi[j-1];
  63.         if (s[i] == s[j])  j++;
  64.         pi[i] = j;
  65.     }
  66.  
  67.     return pi;
  68. }
  69.  
  70. bool occur[50][26];
  71.  
  72. void solve()
  73. {
  74.    int n;
  75.     cin >> n;
  76.  
  77.    string s,t;
  78.  
  79.    cin >> t >> s;
  80.  
  81.    f(i,s.length())
  82.    {
  83.        occur[i][s[i]-'a'] = 1;
  84.        
  85.        if(i > 0)
  86.            f(j,26) occur[i][j]|=occur[i-1][j];
  87.    }  
  88.      
  89.    vector<int> kmp = prefix_function(s);
  90.  
  91.    matrix trans;
  92.  
  93.   for(int i=0;i<t.length();i++)
  94.    {
  95.       for(int k=0;k<s.length();k++)
  96.       {
  97.           if(occur[k][t[i]-'a'] == 0)
  98.           {
  99.               trans.a[0][k]++;
  100.           }
  101.           else if(t[i] == s[k])
  102.           {
  103.               if(k+1 != s.length()) trans.a[k+1][k]++;
  104.           }
  105.           else if(k == 0) trans.a[k][k]++;
  106.           else
  107.           {
  108.                 int j = kmp[k-1];
  109.                 while (j > 0 && t[i] != s[j])  j = kmp[j-1];
  110.                 if (t[i] == s[j])  j++;
  111.                 if(j != s.length()) trans.a[j][k]++;
  112.           }
  113.       }
  114.  
  115.    }
  116.  
  117.    trans = pow(trans,n);
  118.  
  119.    int res = 0;
  120.  
  121.    f(i,SZ) res = ( res + trans.a[i][0]);
  122.    res&=MOD;
  123.  
  124.    cout << res << '\n';      
  125. }
  126.  
  127. signed main()
  128. {
  129.     fast;
  130.  
  131.     int t = 1;
  132.  
  133.     cin >> t;
  134.  
  135.     for(int i=1;i<=t;i++)    
  136.     {
  137.      cout <<"Case "<<i<<": ";
  138.         solve();
  139.     }
  140. }
Add Comment
Please, Sign In to add comment