[luogu2461 SDOI2008] 递归数列 (矩阵乘法)
Description
一个由自然数组成的数列按下式定义:
对于i <= k:ai = bi
对于i > k: ai = c1ai-1 + c2ai-2 + ... + ckai-k
其中bj 和 cj (1<=j<=k)是给定的自然数。写一个程序,给定自然数m <= n, 计算am + am+1 + am+2 + ... + an, 并输出它除以给定自然数p的余数的值。
Input
输入文件spp.in由四行组成。
第一行是一个自然数k。
第二行包含k个自然数b1, b2,...,bk。
第三行包含k个自然数c1, c2,...,ck。
第四行包含三个自然数m, n, p。
Output
输出文件spp.out仅包含一行:一个正整数,表示(am + am+1 + am+2 + ... + an) mod p的值。
Sample Input
2
1 1
1 1
2 10 1000003
Sample Output
142
HINT
对于100%的测试数据:
1<= k <=15
1 <= m <= n <= 1018
对于20%的测试数据:
1<= k <=15
1 <= m <= n <= 106
对于30%的测试数据:
k=1 1 <= m <= n <= 1018
对于所有测试数据:
0<= b1, b2,... bk, c1, c2,..., ck<=109
1 <= p <= 108
Solution
构造矩阵然后直接乘
注意乘的顺序还有最后一定要再MOD一次以防万一qwq
Code
//By Menteur_Hxy
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#define F(i,a,b) for(register int i=(a);i<=(b);i++)
using namespace std;
typedef long long LL;
LL read() {
LL x=0,f=1; char c=getchar();
while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}
while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
return x*f;
}
const int K=20;
int k;
int c[K],b[K];
LL n,m,MOD,ans1,ans2,s[K];
struct Matrix{
LL da[K][K];
Matrix() {clear();};
void clear() {memset(da,0,sizeof(da));}
Matrix operator * (const Matrix oth) {
Matrix res;
F(i,0,k) F(j,0,k) F(l,0,k)
res.da[i][j]=(res.da[i][j]+oth.da[i][l]*da[l][j]%MOD)%MOD;
return res;
}
void print() {
F(i,0,k) {
F(j,0,k) cout<<da[i][j]<<" ";
cout<<endl;
}cout<<endl;
}
}ans,st;
Matrix Qpow(Matrix a,LL d) {
Matrix ret;
ret.da[0][0]=s[k];
F(i,1,k) ret.da[i][0]=b[k-i+1];
while(d) {
if(d&1) ret=ret*a;
a=a*a; d>>=1;
// a.print();
}
return ret;
}
int main() {
k=read();
F(i,1,k) b[i]=read(),s[i]=s[i-1]+b[i];
F(i,1,k) c[i]=read();
m=read(),n=read(),MOD=read();
st.da[0][0]=1;
F(i,1,k) st.da[0][i]=st.da[1][i]=c[i];
F(i,2,k) st.da[i][i-1]=1;
// st.print();
if(n<=k) {
printf("%lld",s[n]-s[m-1]);
return 0;
} else {
ans=Qpow(st,n-k);
// ans.print();
ans1=ans.da[0][0];
}
if(m<=k) ans2=s[m-1];
else {
ans=Qpow(st,m-k-1);
// ans.print();
ans2=ans.da[0][0];
}
printf("%lld",(ans1-ans2+MOD)%MOD);
return 0;
}