1126 求递推序列的第N项

 

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

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

收起

输入

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

输出

输出f(n)的值。

输入样例

3 -1 5

输出样例

6

这道题刚开始打算用暴力写一下结果发现，问题出现在负数上面了，然后打表的时候就得先将其变成整数，然后取余求和，刚开始我写的是到一万的中间过程判断，然后超时好几次，改来改去就改到了50然后数据就通过了，其实我看来的时间复杂度都差不多。

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
int s[55];
int main()
{
    int  a,b,n,i;
    scanf("%d%d%d",&a,&b,&n);
    memset(s,0,sizeof(s));
    s[1]=1;
    s[2]=1;
    for(i=3; i<=49; i++)
    {
        s[i]=(a*s[i-1]+b*s[i-2])%7+7;
        s[i]=s[i]%7;
        if(s[i-1]==1&&s[i]==1)
        {
            break;
        }
    }
    i-=2;
    s[0]=s[i];
    printf("%d\n",s[n%i]);
    return 0;
}