Advertisement
Saleh127

Spoj FIBTWIST / Mat Expo

Feb 21st, 2022
772
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.49 KB | None | 0 0
  1. /***
  2.  created: 2022-02-21-21.53.20
  3. ***/
  4.  
  5. #include <bits/stdc++.h>
  6. using namespace std;
  7. #define ll long long
  8. #define test int tt; cin>>tt; for(int cs=1;cs<=tt;cs++)
  9. #define get_lost_idiot return 0
  10. #define nl '\n'
  11. #define row vector<ll>
  12. #define matrix vector<vector<ll>>
  13. ll mod=1e9+7;
  14.  
  15. matrix operator*(const matrix& a, const matrix& b)
  16. {
  17.     ll r,c,p;
  18.  
  19.     r=a.size();
  20.     p=a[0].size();
  21.     assert(p==b.size()); ///checking row==col
  22.  
  23.     /// first matrix must have the same number of columns
  24.     ///as the second matrix has rows
  25.  
  26.     c=b[0].size();
  27.  
  28.     matrix ans(r,row(c,0));
  29.  
  30.     ll i,j,k;
  31.  
  32.     for(i=0; i<r; i++)
  33.     {
  34.         for(j=0; j<c; j++)
  35.         {
  36.             for(k=0; k<p; k++)
  37.             {
  38.                 ans[i][j]=(ans[i][j]+a[i][k]*b[k][j])%mod;
  39.             }
  40.         }
  41.     }
  42.     return ans;
  43. }
  44.  
  45. matrix bigmod(matrix &a,ll p)
  46. {
  47.     if(p==1) return a;
  48.  
  49.     matrix x=bigmod(a,p/2);
  50.  
  51.     x=x*x;
  52.  
  53.     if(p&1) x=(x*a);
  54.  
  55.     return x;
  56. }
  57.  
  58. int main()
  59. {
  60.     ios_base::sync_with_stdio(0);
  61.     cin.tie(0);
  62.     cout.tie(0);
  63.  
  64.  
  65.     matrix base = {{1,1,1,0},{1,0,0,0},{0,0,1,1},{0,0,0,1}};
  66.     matrix statem = {{1},{0},{1},{1}};
  67.  
  68.     test
  69.     {
  70.         ll n,m;
  71.  
  72.         cin>>n>>mod;
  73.  
  74.         if(n<=1)
  75.         {
  76.             n=n;
  77.         }
  78.         else
  79.         {
  80.             matrix ans= bigmod(base,n-1);
  81.  
  82.             ans=ans*statem;
  83.  
  84.             n=ans[0][0];
  85.         }
  86.        
  87.         cout<<n%mod<<nl;
  88.  
  89.     }
  90.  
  91.     get_lost_idiot;
  92. }
  93.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement