Number Sequence(矩阵快速幂)

网上有流传49为一个周期循环,但经过实际打表,发现有部分数据不对(但可以ac),所以应该用矩阵快速幂求解
关系矩阵:
A B
1 0

#include<iostream>
using namespace std;
struct M{
	int m[2][2];
}; 
int A,B;
M mul(M a,M b)
{
	M ans;
	for(int i=0;i<2;i++)
	{
		for(int j=0;j<2;j++)
		{
			ans.m[i][j] = 0;
			for(int k=0;k<2;k++) ans.m[i][j] = (ans.m[i][j]+a.m[i][k]*b.m[k][j]%7)%7;
		}
	}
	return ans;
}
M quick(long long b)
{
	M ans,c;
	ans.m[0][0] = ans.m[1][1] = 1;
	ans.m[0][1] = ans.m[1][0] = 0;
	c.m[0][0] = A,c.m[0][1] = B,c.m[1][0] = 1,c.m[1][1] = 0;
	while(b)
	{
		if(b&1) ans = mul(ans,c);
		b >>= 1;
		c = mul(c,c);
	}
	return ans;
}
int main()
{
	long long n;
	while(cin>>A>>B>>n)
	{
		if(A==0&&B==0&&n==0) break;
		if(n<=2) cout<<"1"<<endl;
		else
		{
			M a = quick(n-2);
			cout<<(a.m[0][0]+a.m[0][1])%7<<endl;
		}
	}
}
发布了22 篇原创文章 · 获赞 3 · 访问量 638

猜你喜欢

转载自blog.csdn.net/qq_45520541/article/details/105023420