矩阵乘法优化线性递推

对于一个线性递推：f[n]=a1*f[n-1]+a2*f[n-2]+a3*f[n-3]+..+ak*f[n-k]，如何在O(k^3*log(n))的时间复杂度内计算出f[n]

我们需要构造一个k*k的转移矩阵a                                     我们再构造一个状态矩阵b

0        1        0        0        0 ...... 0                                                       f[1]

0        0        1        0        0 ...... 0                                                       f[2]

0        0        0        1        0 ...... 0                                                       f[3]

....................................................                                                        ...

0        0        0        0        1...... 0                                                       f[k-2]

0        0        0        0        0 ...... 1                                                      f[k-1]

a1     a2      a3      a4      a5 ......ak                                                      f[k]

对于每次a*b就相当于整体向前推了一位，则答案就是b*a^(n-k)

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
using namespace std;
const int mod=7777777;
const int N=20;
struct mat{int m[N][N];}a,b,e;
int k,n;
mat mul(mat x,mat y){
	mat c;rep(i,1,k)rep(j,1,k)c.m[i][j]=0;
	rep(i,1,k)rep(j,1,k)rep(p,1,k)c.m[i][j]=(c.m[i][j]+1LL*x.m[i][p]*y.m[p][j]%mod)%mod;
	return c;
}
mat power(mat a,int b){
	mat ans; rep(i,1,k)rep(j,1,k)ans.m[i][j]=i==j;
	for(;b;b>>=1,a=mul(a,a))if(b&1)ans=mul(ans,a);
	return ans;
}
int main()
{
	scanf("%d%d",&k,&n);
	rep(i,1,k-1)a.m[i][i+1]=1; rep(i,1,k)a.m[k][i]=1;
	b.m[0][1]=1;rep(i,1,k)rep(j,0,i-1)b.m[i][1]+=b.m[j][1];
	if(n<=k){cout<<b.m[n][1];return 0;}
	a=power(a,n-k);
	b=mul(a,b);
	cout<<b.m[k][1];
	return 0;
}