ZZULIOJ 1605: 数字序列 (矩阵快速幂）

时间限制: 1 Sec  内存限制: 60 MB
提交: 994  解决: 236
[提交] [状态] [讨论版] [命题人:541307010108]

题目描述

一个数列的定义如下：

f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.

给出A和B，你要求出f(n).

输入

输入包含多个测试案例。每个测试用例包含3个整数A，B和n在一行（1＜＝A，B≤1000，1≤n≤100000000）。

当输入三个0表示结束

输出

对于每个测试案例，输出f(n)，单独占一行。

样例输入 Copy

1 1 3
1 2 10
0 0 0

样例输出 Copy

2
5

来源/分类

 

[提交] [状态]

看这个n的规模，矩阵快速幂；

A B      ×      F2        ＝       A * F2 + B * F1        =      F3

1 0               F1                        F2                               F2

因为乘一次得到的即是第三项，所以求第n项 就是乘 n   - 2 次；

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int A, B;
struct M{
	int m[3][3];
};
M mul(M a, M b)
{
	M ans;
	for (int i = 1; i <= 2; i++)
		for (int j = 1; j <= 2; j++)
		{
			ans.m[i][j] = 0;
			for (int k = 1; k <= 2; k++)
				ans.m[i][j] = (ans.m[i][j] + a.m[i][k] * b.m[k][j] % 7) % 7;
		}
	return ans;
}
ll fpow(int n)
{
	M a, b;
	a.m[1][1] = A;
	a.m[1][2] = B;
	a.m[2][1] = 1;
	a.m[2][2] = 0;
	b.m[1][1] = b.m[2][2] = 1;
	b.m[1][2] = b.m[2][1] = 0;
	while (n)
	{
		if (n & 1)
			b = mul(b, a);
		n >>= 1;
		a = mul(a, a);
	}
	return (b.m[1][1] + b.m[1][2]) % 7;
}
int main(){
	int n;
	while (cin >> A >> B >> n, A, B, n)
	{
		ll ans = 0;
		if (n <= 2)
			ans = -1;
		else
		{
			n -= 2;
			ans = fpow(n);
		}
		cout << ans << endl;
	}
	return 0;
}