(VIJOS) VOJ 1067 Warcraft III 守望者的烦恼 矩阵快速幂

https://vijos.org/p/1067

 

就..挺普通的一道题..自己学一下怎么推式子就可以...细节不多但是我还是日常爆细节..比如说循环写成从负数开始...

 

只求ac不求美观的丑陋的代码....

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<queue>
 using namespace std;
 const int maxn=;
 const double eps=1e-;
 const long long modn=;
 int n,m;
 long long f[]={};
 struct mat{
     long long e[][];
 };
 mat pro(mat x,mat y){
     mat z;
     for(int i=;i<=n;i++){
         for(int j=;j<=n;j++){
             z.e[i][j]=;
             for(int w=;w<=n;w++){
                 z.e[i][j]+=x.e[i][w]*y.e[w][j];
                 z.e[i][j]%=modn;
             }
         }
     }
     return z;
 }
 mat pow(mat x,int k){
     mat z;
     bool f=;
     while(k){
         if(k%==){
             if(f){
                 z=x;f=;
             }
             else{
                 z=pro(x,z);
             }
         }
         k/=;
         x=pro(x,x);
     }
     return z;
 }
 int main(){
     while(~scanf("%d%d",&n,&m)){
         f[]=;
         for(int i=;i<=n;i++){
             f[i]=;
             for(int j=;j<i;j++){
                 f[i]+=f[j];
             }
             f[i]%=modn;
         }
         if(m>n){
             mat a;
             for(int i=;i<=n;i++){
                 for(int j=;j<=n;j++){
                     a.e[i][j]=;
                 }
             }
             for(int i=;i<n;i++){
                 a.e[i][i+]=;
                 a.e[n][i]=;
             }a.e[n][n]=;
             a=pow(a,m-n+);
             long long ans=;
             for(int i=;i<=n;i++){
                 ans+=a.e[n][i]*f[i-];
                 ans%=modn;
             }
             printf("%lld\n",ans);
         }
         else{
             cout<<f[m]<<endl;
         }
     }
     return ;
 }