51Nod-1126 求递推序列的第N项

1126 求递推序列的第N项
基准时间限制：1 秒 空间限制：131072 KB 分值: 10 难度：2级算法题
有一个序列是这样定义的：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>
using namespace std;
int main(int argc, char const *argv[])
{
    int a,b,n,i;
    int f[10010];
    f[1]=1;
    f[2]=1;
    cin>>a>>b>>n;
    for(i=3;i<300;i++){
        f[i]=((a*f[i-1]+b*f[i-2])%7+7)%7;
        if(f[i]==1&&f[i-1]==1){break;}
    }
    f[0]=f[i-2];
    n=n%(i-2);
    cout<<f[n]<<endl;
    return 0;
}