nyoj 301 递推求值(矩阵递推)

题意：自己看

思路：矩阵递推   推荐一篇文章

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

#define A 1000007
int n=3;
struct node{
	int ke[5][5];
};
node a,su,unit;

void init()
{
	for(int i=0;i<n;i++)
	{
		unit.ke[i][i]=1;
		if(i!=0)
			a.ke[i][2]=1;
		else
			a.ke[i][1]=1;
	}
}
void print(node t)
{
	for(int i=0;i<n;i++){
		for(int j=0;j<n;j++)
			printf("%d ",t.ke[i][j]);
		printf("\n");
	}
}
node mul(node b,node c)
{
	node temp;
	for(int i=0;i<n;i++)
		for(int j=0;j<n;j++)
		{
			temp.ke[i][j]=0;
			for(int h=0;h<n;h++)
				temp.ke[i][j]+=((long long)b.ke[i][h]*c.ke[h][j]+A)%A;
			temp.ke[i][j]=(temp.ke[i][j]+A)%A;
		}
	return temp;
}
node get_sum(int k)
{
	node p=a,q=unit;
	while(k!=1)
	{
		if(k&1)
		{
			k--;
			q=mul(q,p);
		}
		else
		{
			k/=2;
			p=mul(p,p);
		}
	}
	return mul(p,q);
}

int main()
{
	int n,t;
	init();
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d%d%d%d%d",&su.ke[0][0],&su.ke[1][0],&a.ke[1][0],&a.ke[1][1],&su.ke[2][0],&n);
		if(n<=2)printf("%d\n",(su.ke[n-1][0]+A)%A);
		else{
			node p=mul(get_sum(n-2),su);
			printf("%d\n",(p.ke[1][0]+A)%A);
		}
	}
}