hdu5015 233 Matrix(矩阵快速幂)

题目

给你一个矩阵,

第一行是233,233,2333,

然后会给你a[1][0],a[2][0],…,a[n][0]

且a[i][j]=a[i-1][j]+a[i][j-1]%mod

求a[n][m]

思路来源

https://www.cnblogs.com/whatbeg/p/3971994.html

题解

把第一行都左移一位就可以对系数矩阵快速幂了

然后最后a[n]这一行的向量就是答案

注意那个系数3要放在基础向量里,放在系数矩阵里会越乘越大

hdu5015 233 Matrix(矩阵快速幂)

除第一行和最后一行外,其余地方对应对角线及其以左的值都配成1,

心得

本来是挺简单的一个题,考察基本功

然而自己磨蹭了1.5h才搞出来, 还是刷题量不够吧

矩阵快速幂的板子不需要太长,一个基本的就够了

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<map>
using namespace std;
typedef long long ll;
const int maxn=12;
const int mod=1e7+7;
int n,m;
ll a[maxn]; 
struct mat
{
	ll a[maxn][maxn];
	mat()
	{
		memset(a,0,sizeof a);
	}
};
mat operator*(mat a,mat b)
{
	mat c;
	for(int i=0;i<=n+1;++i)
	{
		for(int j=0;j<=n+1;++j)
		{
			for(int k=0;k<=n+1;++k)
			{
				c.a[i][j]+=a.a[i][k]*b.a[k][j]%mod;
				if(c.a[i][j]>=mod)c.a[i][j]%=mod;
			}
		}
	}
	return c;
}
mat modpow(mat x,int p)
{
	mat res;
	for(int i=0;i<=n+1;++i)
	{
		for(int j=0;j<=n+1;++j)
		if(i==j)res.a[i][j]=1;
	}
	for(;p;x=x*x,p/=2)
	if(p&1)res=res*x; 
	return res;
} 
int main()
{
	while(~scanf("%d%d",&n,&m))
	{
		for(int i=1;i<=n;++i)
		scanf("%lld",&a[i]);
		a[0]=233;a[n+1]=3;
		if(n==0&&m==0)
		{
			puts("0");
			continue;
		}
		//if(!n)m--;
		mat ans;
		ans.a[0][0]=10;
		ans.a[0][n+1]=1;
		ans.a[n+1][n+1]=1;
		for(int i=1;i<=n;++i)
		{
			for(int j=0;j<=i;++j)
			ans.a[i][j]=1;
		}
		ans=modpow(ans,m);
		ll res=0;
		for(int i=0;i<=n+1;++i)
		{
		 res+=ans.a[n][i]*a[i]%mod;
		 if(res>=mod)res%=mod;
	    }
	    printf("%lld\n",res);
	}
	return 0;
}