CodeForces - 450B Jzzhu and Sequences

最新推荐文章于 2020-09-19 19:07:20 发布

M_Lter

最新推荐文章于 2020-09-19 19:07:20 发布

阅读量134

点赞数

分类专栏：矩阵数论文章标签：快速矩阵幂

本文链接：https://blog.csdn.net/lidengdengter/article/details/81479410

版权

数论同时被 2 个专栏收录

51 篇文章 0 订阅

订阅专栏

矩阵

13 篇文章 0 订阅

订阅专栏

题目链接

给定前两个值x，y，根据递推式求第n项的值。 $f_{i}=f_{i-1}+f_{i+1}. i\geq 2$ ==》 $f_{i+1}=f_{i}-f_{i-1}$

即递推式整理为f(n)=f(n-1)-f(n-2) ,n>=2,因为数大，用快速矩阵幂求解，套模板即可

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
typedef long long ll;
const ll mod=1e9+7;
struct matrix{
	ll x[2][2];
};
matrix multi(matrix a,matrix b)//矩阵相乘
{
	matrix temp;
	memset(temp.x,0,sizeof(temp.x));
	for(int i=0;i<2;i++)
		for(int j=0;j<2;j++)
			for(int k=0;k<2;k++)
			{
				temp.x[i][j]+=a.x[i][k]*b.x[k][j];
				temp.x[i][j]%=mod;
			}
	return temp;
}
matrix quick_multi(matrix a,ll n)//矩阵快速幂
{
	matrix temp=a;
	n--;
	while(n){
		if(n&1)
			temp=multi(temp,a);
		a=multi(a,a);
		n>>=1;
	}
	return temp;
}
int main()
{
	ll x,y,n;
	while(~scanf("%lld%lld",&x,&y))
	{
		if(x<0)
			x+=mod;
		if(y<0)
			y+=mod;
		scanf("%lld",&n);
		if(n==1){
			printf("%lld\n",x%mod);
			continue;
		}
		if(n==2){
			printf("%lld\n",y%mod);
			continue;
		}
		
		x%=mod;y%=mod;
		matrix A;
		matrix ans;
		memset(A.x,0,sizeof(A.x));
		memset(ans.x,0,sizeof(ans.x));
		A.x[0][0]=1;A.x[1][0]=-1;
		A.x[0][1]=1;A.x[1][1]=0;
		
		ans.x[0][0]=y,ans.x[0][1]=x;
		
		A=quick_multi(A,n-2);
		ans=multi(ans,A);
		printf("%lld\n",(ans.x[0][0]+mod)%mod);
	}
 
	return 0;
}