由于n太大,不能直接递推,需要用矩阵快速幂来解决,时间复杂度为O(d^3logn)
举例,d=5的矩阵关系式为:
|a1 a2 a3 a4 a5| | f[n-1] | | f[n]|
|1 | | f[n-2] | | f[n-1] |
| 1 | * | f[n-3] | = | f[n-2] | (空白处为0)
| 1 | | f[n-4] | | f[n-3] |
| 1 | | f[n-5] | | f[n-4] |
F(n)=A*F(n-1);
所以 F(n)=A^(n-d)*F(d);
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=20;
struct Mat{
ll mat[N][N];
}sta,fi;
ll d,mod;
ll n,m;
Mat mul(Mat a,Mat b){
Mat temp;
memset(temp.mat,0,sizeof(temp.mat));
for(int i=1;i<=d;i++)
for(int j=1;j<=d;j++)
for(int k=1;k<=d;k++)
{
temp.mat[i][j]=(temp.mat[i][j]+(a.mat[i][k]*b.mat[k][j])%m)%m;
}
return temp;
}
Mat quick(ll k){
Mat temp;
memset(temp.mat,0,sizeof(temp.mat));
for(int i=1;i<=20;i++)
temp.mat[i][i]=1;
while(k){
if(k&1)
temp=mul(temp,sta);
k>>=1;
sta=mul(sta,sta);
}
return temp;
}
int main(){
while(~scanf("%d %lld %lld",&d,&n,&m)){
memset(sta.mat,0,sizeof(sta.mat));
memset(fi.mat,0,sizeof(sta.mat));
for(int i=2;i<=d;i++)
sta.mat[i][i-1]=1;
for(int i=1;i<=d;i++){
scanf("%lld",&sta.mat[1][i]);
sta.mat[1][i]%=m;
}
for(int i=1;i<=d;i++){
scanf("%lld",&fi.mat[d-i+1][1]);
fi.mat[d-i+1][1]%=m;
}
if(n<=(ll)d){
printf("%lld\n",fi.mat[d-n+1][1]);
continue;
}
fi=mul(quick(n-d),fi);
printf("%lld\n",fi.mat[1][1]);
}
}