http://acm.nyist.edu.cn/JudgeOnline/problem.php?pid=301 (矩陣的用法)

#include <iostream>
using namespace std;
long long f1,f2,e,g,cc,n,a[3][3],b[3][3],c[3][3];
long long f()
{
	if(n==1) return f1;    if(n==2) return f2;
	long long flag=0,m,k=0;
	a[0][0]=g,a[0][1]=1,a[0][2]=0;
	a[1][0]=e,a[1][1]=0,a[1][2]=0;
	a[2][0]=1,a[2][1]=0,a[2][2]=1;
    n-=2;
	while(n)
	{
		if(n%2&&flag==1)
		{
			for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			{c[i][j]=0;
			for(int r=0;r<3;r++)
				c[i][j]+=b[i][r]*a[r][j];}
			for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			b[i][j]=c[i][j]%1000007;
		} 
		if(n%2&&flag==0) 
		{
			for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			b[i][j]=a[i][j];
	     flag=1;
		}
	        for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			{c[i][j]=0;
			for(int r=0;r<3;r++)
				c[i][j]+=a[i][r]*a[r][j];}
			for(int i=0;i<3;i++)
			for(int j=0;j<3;j++)
			a[i][j]=c[i][j]%1000007;
	     n/=2;k++;
	}  
    m=f2*b[0][0]+f1*b[1][0]+cc*b[2][0]; m=m%1000007;
	return (m+1000007)%1000007;
}
int main(int argc, char *argv[])
{
	int t;
	cin>>t;
	while(t--)
	{
		cin>>f1>>f2>>e>>g>>cc>>n;
		cout<<f()<<endl;
	}
	return 0;
}

發佈了54 篇原創文章 · 獲贊 2 · 訪問量 4萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章