Advertisement
Saleh127

UVA 10229 / Mat Exppo

Feb 21st, 2022
1,229
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.44 KB | None | 0 0
  1. /***
  2.  created: 2022-02-21-11.40.15
  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;
  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.     matrix base = {{1,1},{1,0}};
  65.     matrix statem = {{1},{0}};
  66.  
  67.     ll n,m;
  68.  
  69.     while(cin>>n>>m)
  70.     {
  71.         mod=(1<<m);
  72.  
  73.         if(n<=1)
  74.         {
  75.             cout<<n%mod<<nl;
  76.         }
  77.         else
  78.         {
  79.             matrix ans = bigmod(base,n-1);
  80.  
  81.             ans=ans*statem;
  82.  
  83.             cout<<ans[0][0]%mod<<nl;
  84.         }
  85.     }
  86.  
  87.     get_lost_idiot;
  88. }
  89.  
  90.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement