uva 10870 Recurrences(矩阵优化递推)

最新推荐文章于 2018-09-04 17:32:09 发布

yjCola

最新推荐文章于 2018-09-04 17:32:09 发布

阅读量502

点赞数

分类专栏：递推/矩阵文章标签：矩阵优化递推

本文链接：https://blog.csdn.net/fallen_fall/article/details/24588813

版权

递推/矩阵专栏收录该内容

21 篇文章 0 订阅

订阅专栏

uva 10870 Recurrences

比较裸的矩阵，注意数据给出的顺序

#include<stdio.h>
#include<string.h>
#define MAXN 20
#define ll long long
struct mat
{
	ll m[MAXN][MAXN];
};
int n,mod;
mat multi(const mat &a,const mat &b)
{
	mat ans;
	memset(&ans,0,sizeof(ans));
	for(int i=0;i<n;i++)
		for(int k=0;k<n;k++)
			if(a.m[i][k])
				for(int j=0;j<n;j++)
				{
					ans.m[i][j]+=a.m[i][k]*b.m[k][j];
					ans.m[i][j]%=mod;
				}
	return ans;
}
mat pow(mat a,int k)
{
	mat ans;
	memset(&ans,0,sizeof(ans));
	for(int i=0;i<n;i++) ans.m[i][i]=1;
	while(k)
	{
		if(k&1) ans=multi(ans,a);
		a=multi(a,a);
		k>>=1;
	}
	return ans;
}
int b[MAXN];
int main()
{
	mat a;
	int k;
	while(scanf("%d%d%d",&n,&k,&mod)!=EOF&&(n+k+mod))
	{
		memset(&a,0,sizeof(a));
		for(int i=0;i<n;i++) scanf("%lld",&a.m[0][i]);
		for(int i=n-1;i>=0;i--) scanf("%d",&b[i]);

		if(k<=n) printf("%d\n",b[n-k]%mod);
		else
		{
			for(int i=1;i<n;i++) a.m[i][i-1]=1;
			a=pow(a,k-n);
			ll ans=0;
			for(int i=0;i<n;i++)
			{
				ans+=a.m[0][i]*b[i];
				ans%=mod;
			}
			printf("%lld\n",ans);
		}
	}
	return 0;
}