[Vijos 1067] Warcraft III 守望者的烦恼 · 矩阵乘法

矩阵乘法。

首先，0~k-1这一段的方案要预处理出来（别问我为什么）。

之后用DP做的话递推式很简单：

,但是N<=2^31-1，所以这样做是肯定会爆的~

那么我们对于这个递推式，我们很容易可以建立一个k*k的矩阵：

f[i+1][i]=1(i=1~k-1),f[1..k][k]=1,例如k=4时：

Matrix=
这时我们用一开始预处理的[f0 f1 f2 .. fk-2 fk-1]*Matrix，得到的是[f1 f2 f3 .. fk-1 fk]

这是一次递推的矩阵，那么我们一开始已经预处理了k-1次，剩下的n-k+1次就妥妥的用快速幂水过去啦~~

另外蒟蒻笔者一开始还不知道什么叫单位矩阵，在此介绍：

单位矩阵的左上-右下对角线上全为1，其余为0，设单位矩阵为P，对于任意能与P相乘的矩阵Q，总有P*Q=Q，即Q不变。

#include <stdio.h>
#include <algorithm>
#include <string.h>
#include <iostream>
using namespace std;

const int p=7777777;
struct matrix{
	long long n,m,num[15][15];
	void init(int i,int j){
		n=i;m=j;
		memset(num,0,sizeof num);
	}
};
int n,k;
matrix ans;

matrix operator *(matrix a,matrix b){
	matrix c;
	c.init(a.n,b.m);
	for (int i=1;i<=c.n;i++)
		for (int j=1;j<=c.m;j++)
			for (int t=1;t<=a.m;t++)
				c.num[i][j]=(c.num[i][j]+a.num[i][t]*b.num[t][j])%p;
	return c;
}

matrix work(){
	matrix unit,ans;
	unit.init(k,k);
	for (int i=1;i<k;i++) unit.num[i+1][i]=1;
	for (int j=1;j<=k;j++) unit.num[j][k]=1;
	ans.init(k,k);
	for (int i=1;i<=k;i++) ans.num[i][i]=1;<span style="white-space:pre">	</span>//ans一开始是一个单位矩阵
	
	for (int tot=n-k+1;tot;tot>>=1){
		if (tot&1) ans=ans*unit;
		unit=unit*unit;
	}
	return ans;
}

int main(){
	cin>>k>>n;ans.init(1,k);
	ans.num[1][1]=1;
	for (int i=2;i<=k;i++)
		for (int j=1;j<i;j++)
			ans.num[1][i]=(ans.num[1][i]+ans.num[1][j])%p;
	ans=ans*work();
	cout<<ans.num[1][k];
	return 0;
}