题意:f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.给出A,B,n求f(n)。 (1 <= A, B <= 1000, 1 <= n <= 100,000,000),三个都是0的时候停止。
【代码1:MLE】
#include <iostream>
using namespace std;
#define MAX 100000000+2
int num[MAX];
int main()
{
int A = 0, B = 0, n = 0;
num[0] = 1, num[1] = 1;
while (cin >> A >> B >> n && (A+B+n)) {
int i = 0;
for (i = 2; i < n; i++)
num[i] = (A*num[i-1] + B*num[i-2])%7;
cout << num[n-1] <<endl;
}
return 0;
}
1亿的数组内存超限,可以知道结果必定存在周期,问题在于怎样找到这个周期。
f(n-1),f(n-2)都有7种可能,所以周期最大为49。所以循环49次按照49来取就可以。
【代码2:AC】
#include <iostream>
using namespace std;
#define MAX 50+2
int num[MAX];
int main()
{
int A = 0, B = 0, n = 0;
num[1] = 1, num[2] = 1;
while (cin >> A >> B >> n && (A+B+n))
{
int i = 0;
for (i = 3; i < 50; i++)
num[i] = (A*num[i-1] + B*num[i-2])%7;
n = n%49;
if (n == 0)
cout << num[49] << endl;
else
cout <<num[n] << endl;
}
return 0;
}