比较简单的一道找规律题,一开始没优化,果然超内存了...
对于公式 f[n] = A * f[n-1] + B * f[n-2] 只有7 * 7 = 49 种可能
因为对于f[n-1] 或者 f[n-2] 的取值只有 0,1,2,3,4,5,6 这7个数,A,B又是固定的,所以就只有49种可能值了
由该关系式得知每一项只与前两项发生关系,所以当连续的两项在前面出现过循环节出现了,注意循环节并不一定会是开始的 1,1
又因为一组测试数据中f[n]只有49中可能的答案,最坏的情况是所有的情况都遇到了,那么那也会在50次运算中产生循环节
找到循环节后,就OK啦,内存也优化啦~
代码如下:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long ll;
const ll N = 1e8+10;
int n, a, b, s, e;
int f[60];
int main()
{
f[0] = f[1] = f[2] = 1;
while(~scanf("%d%d%d", &a, &b, &n))
{
if(a == 0 && b == 0 && n == 0) break;
s = e = 0;
int flag = 0;
for(int i = 3; i <= n && !flag; i++)
{
f[i] = (a * f[i - 1] + b * f[i - 2]) % 7;
for(int j = 2; j <= i-1; j++)
{
if(f[j] == f[i] && f[j-1] == f[i-1])
{
s = j + 1, e = i + 1; flag = 1; break;
}
}
}
if(flag)
{
printf("%d\n", f[s + (n - e)%(e - s)]);
}
else
printf("%d\n", f[n]);
}
return 0;
}