51nod 1126 求递推序列的第N项

题目链接：https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1126

题目：

有一个序列是这样定义的：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)的值。

由于f[n]只可能在0到6之间。所以前后两个数一共有49种可能。只需要找循环节即可。

#include <iostream>
#include<bits/stdc++.h>
#define N 1100000000
using namespace std;

int f[11000];
int mp[7][7];

int main()
{
    int a,b,n;
    scanf("%d%d%d",&a,&b,&n);
    f[1]=f[2]=1;
    memset(mp,0,sizeof(mp));
    mp[1][1]=1;
    int st=0,ed=N;
    for(int i=3;i<=n;i++)
    {
        f[i]=((a*f[i-1]+b*f[i-2])%7+7)%7;
        if(mp[f[i-1]][f[i]])
        {
            st=mp[f[i-1]][f[i]];
            ed=i-1;
            break;
        }
        mp[f[i-1]][f[i]]=i-1;
    }
    cout<<f[(n-st)%(ed-st)+st]<<endl;
}