HDOJ 1005 Number Sequence

题意：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;
}