Saleh127

Spoj FIBOSUM / Mat Expo

Feb 19th, 2022
1,267
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.86 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+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,1},{0,1,0}};
  66.     matrix statem = {{1},{1},{0}};
  67.  
  68.     test
  69.     {
  70.         ll n,m;
  71.  
  72.         cin>>n>>m;
  73.  
  74.         if(n>0) n--;
  75.  
  76.         if(n<=2)
  77.         {
  78.             n=n;
  79.         }
  80.         else if(n==3) n=4;
  81.         else
  82.         {
  83.             matrix ans= bigmod(base,n-1);
  84.  
  85.             ans=ans*statem;
  86.  
  87.             n=ans[0][0];
  88.         }
  89.  
  90.         if(m<=2)
  91.         {
  92.             m=m;
  93.         }
  94.         else if(m==3) m=4;
  95.         else
  96.         {
  97.             matrix ans1= bigmod(base,m-1);
  98.  
  99.             ans1=ans1*statem;
  100.  
  101.             m=ans1[0][0];
  102.         }
  103.  
  104.         //cout<<m<<" "<<n<<nl;
  105.  
  106.         ll sol=(m-n);
  107.  
  108.         if(sol<0)
  109.         {
  110.              sol+=mod;
  111.         }
  112.  
  113.         cout<<sol%mod<<nl;
  114.  
  115.     }
  116.  
  117.     get_lost_idiot;
  118. }
  119.  
Advertisement
Add Comment
Please, Sign In to add comment