Advertisement
Saleh127

UVA 12470 / Mat Expo

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