BZOJ 3231: [Sdoi2008]递归数列 (JZYZOJ 1353) 矩阵快速幂

http://www.lydsy.com/JudgeOnline/problem.php?id=3231

 

和斐波那契一个道理在最后加一个求和即可

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 //using namespace std;
 const int maxn=;
 const double eps=1e-;
 long long modn;
 long long n,l,r;
 long long b[]={};
 struct mat{
     long long e[][];
     mat(){ memset(e,,sizeof(e)); }
 };
 mat a;
 mat Mul(mat x,mat y){
     mat z;
     for(int i=;i<=n+;i++){
         for(int j=;j<=n+;j++){
             for(int k=;k<=n+;k++){
                 z.e[i][j]+=x.e[i][k]*y.e[k][j];
                 z.e[i][j]%=modn;
             }
         }
     }
     return z;
 }
 mat Pow(mat x,long long k){
     mat z;
     for(int i=;i<=n+;i++){
         z.e[i][i]=;
     }
     while(k>){
         if(k&){
             z=Mul(z,x);
         }
         x=Mul(x,x);
         k/=;
     }/*for(int i=1;i<=n;i++){
         for(int j=1;j<=n;j++){
             std::cout<<z.e[i][j]<<' ';
         }
         std::cout<<std::endl;
     }*/
     return z;
 }
 long long doit(long long x){
     if(x<n){
         return b[x+];
     }
     mat z=Pow(a,x-n+);
     long long ans=,s=,d=;
     for(int i=;i<=n+;i++){
         d+=z.e[n][i]*b[i];
         s+=z.e[n+][i]*b[i];
         d%=modn;s%=modn;
     }
     ans=(s+d)%modn;
     ans%=modn;
     return ans;
 }
 int main(){
     scanf("%lld",&n);
     n+=;
     for(int i=;i<=n;i++){
         scanf("%lld",&b[i]);
         b[n+]+=b[i];
     }
     b[n+]-=b[n];
     for(int i=n;i>;i--){
         scanf("%lld",&a.e[n][i]);
     }
     long long l,r;
     scanf("%lld%lld%lld",&l,&r,&modn);
     for(int i=;i<=n;i++){
         b[i]%=modn;
         a.e[i-][i]=;a.e[n][i]%=modn;
     }
     a.e[n+][n+]=,a.e[n+][n]=;
     /*for(int i=1;i<=n+1;i++){
         for(int j=1;j<=n+1;j++){
             std::cout<<a.e[i][j]<<' ';
         }
         std::cout<<std::endl;
     }*/
     long long ans=(doit(r)-doit(l-)+modn)%modn;
     printf("%lld\n",ans);
     return ;
 }