51nod1126 求递推序列的第N项【递推】

有一个序列是这样定义的：f(1) = 1, f(2) = 1, f(n) = (A * f(n - 1) + B * f(n - 2)) mod 7.

给出A，B和N，求f(n)的值。

Input

输入3个数：A,B,N。数字之间用空格分割。(-10000 <= A, B <= 10000, 1 <= N <= 10^9)

Output

输出f(n)的值。

Input示例

3 -1 5

Output示例

6

#include<bits/stdc++.h>
 #define N 1000009
using namespace std;
typedef long long LL;
int f[N]; int main()
{
    int i, j, a, b, n, ff = 0;
    f[1] = f[2] = 1;
    scanf("%d%d%d", &a, &b, &n);
    for(i = 3; i <= n; i++)
    {
        f[i] = (((a*f[i-1] + b*f[i-2]) % 7) + 7) % 7;
        for(j = 2; j < i; j++)
        {
            if(f[i] == f[j] && f[i - 1] == f[j - 1])
            {
                ff = 1; break;
            }
        }
        if(ff) break;
    }
    if(ff)
        printf("%d\n", f[j + (n - i) % (i - j)]);
    else
    printf("%d\n", f[n]);
    return 0;
 }