本题的N可能非常大,如果使用普通的递归,则会超时。
观察f(n)的计算式之后,发现f(n)的值的取值范围为{0,1,2,3,4,5,6},又f(n)的值是根据f(n-1)和f(n-2)确定的,所以从第7 * 7 = 49个数,必定回到f(1)开始循环。
#include <iostream>
using namespace std;
int f(int a, int b, int n);
int main()
{
int a, b, n;
while (cin >> a >> b >> n)
{
if (a == 0 && b == 0 && n == 0)
break;
cout << f(a, b, n) << endl;
}
return 0;
}
int f(int a, int b, int n)
{
int arr[50];
arr[1] = arr[2] = 1;
for (int i = 3; i < 50; i++)
{
arr[i] = (a*arr[i - 1] + b*arr[i - 2]) % 7;
}
if (n % 49)
return arr[n % 49];
else
return arr[1];
}
继续加油。