Bzoj4161 Shlw loves matrixI

Time Limit: 80 Sec Memory Limit: 128 MB
Submit: 111 Solved: 55

Description

给定数列 {hn}前k项，其后每一项满足

hn = a1*h(n-1) + a2*h(n-2) + ... + ak*h(n-k)

其中 a1,a2...ak 为给定数列。请计算 h(n)，并将结果对 1000000007 取模输出。

Input

第 1 行包含两个整数 n,k

第 2 行包含 k 个整数 a1,a2...ak

第 3 行包含 k 个整数 h1,h2...hk

Output

一行一个整数 hn mod 1000000007

Sample Input

6 4
3 -1 0 4
-2 3 1 5

Sample Output

HINT

对于 100% 数据，满足 n <= 10^9;k <= 2000; abs(hi)<=10^9; abs(ai)<=10^9

Source

By submittersubmitter

数学问题特征多项式递推

直接矩阵乘法递推的话，$O(k^3 logn)$的复杂度十分难受

我们可以找到递推矩阵的特征多项式，就可以用 $O(k^2)$的卷积代替$O(k^3)$的矩乘，这题就可以过了。数据再丧心病狂的话，还可以用FFT优化卷积。

博主傻傻不会求特征多项式啊……依稀记得在哪里听到过，齐次递推式的递推矩阵的特征多项式和递推式形式一样？真伪不明，反正这题里是一样的。

设矩阵为M，则 $ M^i = sum_{j=1}^{k} a_j * M^{i-j} $

 1 #include<iostream>
 2 #include<algorithm>
 3 #include<cstring>
 4 #include<cstdio>
 5 #include<cmath>
 6 #define LL long long
 7 using namespace std;
 8 const int mod=1e9+7;
 9 const int mxn=4011;
10 int read(){
11     int x=0,f=1;char ch=getchar();
12     while(ch<'0' || ch>'9'){if(ch=='-')f=-1;ch=getchar();}
13     while(ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
14     return x*f;
15 }
16 int n,K;
17 int A[mxn],h[mxn];
18 struct Mat{
19     int x[mxn];
20     void init(){memset(x,0,sizeof x);return;}
21     Mat operator * (const Mat &b){
22         Mat res;
23         for(int i=K<<1;i;i--)res.x[i]=0;
24         for(int i=1;i<=K;i++)
25             for(int j=1;j<=K;j++)
26                 res.x[i+j-1]=((LL)res.x[i+j-1]+(LL)x[i]*b.x[j])%mod;
27         for(int i=K<<1;i>K;i--)
28             for(int j=1;j<=K;j++)
29                 res.x[i-j]=((LL)res.x[i-j]+(LL)res.x[i]*A[j])%mod;
30         return res;
31     }
32 }bas,res;
33 int main(){
34     int i,j;
35     n=read();K=read();
36     for(i=1;i<=K;i++)A[i]=read(),A[i]=(A[i]<0)?A[i]+mod:A[i];
37     for(i=1;i<=K;i++)h[i]=read(),h[i]=(h[i]<0)?h[i]+mod:h[i];
38     bas.x[2]=1;
39     res.x[1]=1;
40     for(i=n;i;i>>=1){
41         if(i&1)res=res*bas;
42         bas=bas*bas;
43     }
44     int ans=0;
45     for(i=1;i<=K;i++){
46         ans=((LL)ans+(LL)res.x[i]*h[i])%mod;
47     }
48     printf("%d
",ans);
49     return 0;
50 }