BZOJ4161 常系数齐次线性递推
生活随笔
收集整理的這篇文章主要介紹了
BZOJ4161 常系数齐次线性递推
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
問了數競的毛毛搞了一番也沒太明白,好在代碼蠻好寫先記下吧。
1 #include<bits/stdc++.h> 2 using namespace std; 3 const int N=4005,mod=1e9+7; 4 int n,k,c[N],b[N],a[N],f[N],tmp[N],ans; 5 inline void qmul(int *x,int *y) 6 { 7 for(int i=0;i<k*2;++i)tmp[i]=0; 8 for(int i=0;i<k;++i) 9 for(int j=0;j<k;++j) 10 tmp[i+j]=(tmp[i+j]+1ll*x[i]*y[j]%mod)%mod; 11 for(int i=k*2-2;i>=k;--i) 12 for(int j=1;j<=k;++j) 13 tmp[i-j]=(tmp[i-j]+1ll*tmp[i]*a[j]%mod)%mod; 14 for(int i=0;i<k;++i)x[i]=tmp[i]; 15 return; 16 } 17 int main() 18 { 19 scanf("%d%d",&n,&k); 20 for(int i=1;i<=k;++i)scanf("%d",&a[i]),a[i]=(a[i]%mod+mod)%mod; 21 for(int i=0;i<k;++i)scanf("%d",&f[i]),f[i]=(f[i]%mod+mod)%mod; 22 c[1]=b[0]=1; 23 if(n<k){ 24 printf("%d\n",f[n]); 25 return 0; 26 } 27 while(n) 28 { 29 if(n&1)qmul(b,c); 30 qmul(c,c);n>>=1; 31 } 32 for(int i=0;i<k;++i)ans=(ans+1ll*f[i]*b[i]%mod)%mod; 33 printf("%d\n",ans); 34 return 0; 35 }?
轉載于:https://www.cnblogs.com/nbwzyzngyl/p/8618360.html
總結
以上是生活随笔為你收集整理的BZOJ4161 常系数齐次线性递推的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 2016012090+小学四则运算练习软
- 下一篇: 个人项目——数独