NYISTME 1300 递推求值

M - 递推求值

Time Limit:1000MS     Memory Limit:65536KB     64bit IO Format:%lld & %llu

Submit  Status  Practice  NYISTME 1300

Description

给你一个递推公式：

f(x)=a*f(x-2)+b*f(x-1)+c

并给你f(1),f(2)的值，请求出f(n)的值，由于f(n)的值可能过大，求出f(n)对1000007取模后的值。

注意:-1对3取模后等于2

Input

第一行是一个整数T,表示测试数据的组数(T<=10000) 随后每行有六个整数，分别表示f(1),f(2),a,b,c,n的值。 其中0<=f(1),f(2)<100,-100<=a,b,c<=100,1<=n<=100000000 (10^9)

Output

输出f(n)对1000007取模后的值

Sample Input

2
1 1 1 1 0 5
1 1 -1 -10 -100 3

Sample Output

5
999896

#include<cstdio>
long long aa[4][4],t[4][4],d[4][4];

int main()
{
	long long f1,f2,a,b,c,n,i,j,k,T;
	scanf("%lld",&T);
	while(T--)
	{
		scanf("%lld%lld%lld%lld%lld%lld",&f1,&f2,&a,&b,&c,&n);
		aa[1][1]=b,aa[1][2]=a,aa[1][3]=c;
		aa[2][1]=1,aa[2][2]=0,aa[2][3]=0;
		aa[3][1]=0,aa[3][2]=0;aa[3][3]=1;
		d[1][1]=f2,d[2][1]=f1,d[3][1]=1;
		if(n==1)
			printf("%lld\n",f1);
		else if(n==2)
			printf("%lld\n",f2);
		else
		{
		n=n-2;
		while(n!=0)
		{
			if(n%2==1)
			{
				for( i=1;i<=3;i++)
				{
					for( j=1;j<=1;j++)
					{
						t[i][j]=0; 
						for( k=1;k<=3;k++)
							t[i][j]+=aa[i][k]*d[k][j];
					}
				}
				for( i=1;i<=3;i++)
					for( j=1;j<=1;j++)
					{
						if(t[i][j]<0)
						d[i][j]=(t[i][j]+1000007)%1000007;
						else
						d[i][j]=t[i][j]%1000007;	
					}
			}
			n/=2;
			{
				if(n==0)
					continue;
				for( i=1;i<=3;i++)
				{
					for( j=1;j<=3;j++)
					{
						t[i][j]=0; 
						for( k=1;k<=3;k++)
							t[i][j]+=aa[i][k]*aa[k][j];
					}
				}
			for(int i=1;i<=3;i++)
				for(int j=1;j<=3;j++)
				{
					if(t[i][j]<0)
					aa[i][j]=(t[i][j]+1000007)%1000007;//这儿要特别注意
					else
					aa[i][j]=t[i][j]%1000007;
				}	
			}
	  	}
		printf("%lld\n",d[1][1]);
		}
    }
	return 0;
}