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

这个题注意取模啊！！！可能是负数，所以要(  (...)+7  )%7

还有这个i<=300有的数据不行，i<200就可以

如果有知道原因的求解答

代码：

#include<iostream>
using namespace std;

int f[300];

int main()
{
    int a,b,n;
    cin>>a>>b>>n;
    f[1]=1;
    f[2]=1;
    if(n==1||n==2)
        cout<<"1"<<endl;
    else
    {
        int i;
        for(i=3;i<300;i++)
        {
            f[i]=((a*f[i-1]+b*f[i-2])%7+7)%7;  //防止负数
            cout<<"f["<<i<<"]="<<f[i]<<endl;
            if(f[i]==1&&f[i-1]==1)
            {
                break;
            }
        }
        f[0]=f[i-2];
        cout<<f[n%(i-2)]<<endl;
    }
    return 0;
}