【HDU 1005】 Number Sequence

【题目链接】

            点击打开链接

【算法】

          矩阵乘法快速幂，即可

【代码】

           

#include<bits/stdc++.h>
using namespace std;

int a,b,n;

struct Matrix
{
		int mat[3][3];
} ans;
inline void multipy(Matrix &a,Matrix b)
{
		int i,j,k;
		Matrix ans;
		ans.mat[1][1] = ans.mat[1][2] = ans.mat[2][1] = ans.mat[2][2] = 0;
		for (i = 1; i <= 2; i++)
		{
				for (j = 1; j <= 2; j++)
				{
						for (k = 1; k <= 2; k++)
						{
								ans.mat[i][j] = (ans.mat[i][j] + a.mat[i][k] * b.mat[k][j]) % 7;
						}
				}
		}
		a = ans;
}
inline Matrix power(Matrix a,int n)
{
		Matrix ans,b = a;
		ans.mat[1][1] = 1; ans.mat[1][2] = 0;
		ans.mat[2][1] = 0; ans.mat[2][2] = 1;
		while (n)
		{
				if (n & 1) multipy(ans,b);
				n >>= 1;
				multipy(b,b);
		}
		return ans;
}
int main() 
{
		
		while (scanf("%d%d%d",&a,&b,&n) != EOF)
		{
				if (!a && !b && !n) break;
				if (n <= 2)
				{
					 printf("%d\n",1);
					 continue;
				}
				ans.mat[1][1] = a; ans.mat[1][2] = b;
				ans.mat[2][1] = 1; ans.mat[2][2] = 0;
				ans = power(ans,n-2);
				printf("%d\n",(ans.mat[1][1]+ans.mat[1][2])%7);		
		}
		
		return 0;
	
}