【题解】洛谷P1349广义斐波那契数列

前往：我自己搭建的博客

题目

洛谷P1349广义斐波那契数列

题解

构造转移矩阵与快速幂结合。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll p,q,a1,a2,n,MOD; 
struct matrix{ll s[3][3];}st,nxt;	//st是初始矩阵，nxt是转移矩阵 
inline matrix mul(matrix a,matrix b)
{
	matrix c; memset(c.s,0,sizeof(c.s));
	for(int i=1;i<=2;i++)
		for(int j=1;j<=2;j++)
			for(int k=1;k<=2;k++)
				c.s[i][j]=(c.s[i][j]+a.s[i][k]*b.s[k][j]%MOD)%MOD;
	return c;
}
inline matrix quick_pow(matrix a,ll b)
{
	matrix c;
	for(int i=1;i<=2;i++)
		for(int j=1;j<=2;j++)
			c.s[i][j]=(i==j)?1:0;
	for( ;b;b>>=1)
	{
		if(b&1) c=mul(c,a);
		a=mul(a,a);
	}
	return c;
}
inline void pre()	//预处理 
{
	nxt.s[1][1]=0;nxt.s[1][2]=1;
	nxt.s[2][1]=q;nxt.s[2][2]=p;
	st.s[1][1]=a1;st.s[2][1]=a2;
}
			
int main()
{
	scanf("%lld%lld%lld%lld%lld%lld",&p,&q,&a1,&a2,&n,&MOD);
	pre(); nxt=quick_pow(nxt,n-1); st=mul(nxt,st); 
	printf("%lld\n",st.s[1][1]);
	
	return 0;
}