题意
有一个序列是这样定义的:f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.
给出A,B和N,求f(n)的值。
解题思路
存在循环节,只需要找到循环节即可。判断是否为循环节的方法是当连着两项为1时,即找到了循环节。
参考代码
#include <iostream>
using namespace std;
int f[300] = {1, 1, 1};
int main(int argc, const char * argv[]){
int A, B, N;
while (cin >> A >> B >> N){
if (N == 1 || N == 2){
printf("1\n");
continue;
}
int i;
for (i = 3; i < 300; i++){
f[i] = ((A * f[i - 1] + B * f[i - 2]) % 7 + 7) % 7;
if (f[i - 1] == 1 && f[i] == 1)
break;
}
f[0] = f[i - 2];
printf("%d\n", f[N % (i - 2)]);
}
return 0;
}