51nod1126求递推序列的第N项和

https://www.51nod.com/onlineJudge/questionCode.html#!problemId=1126&judgeId=582819

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<iostream>
#include<string.h>
using namespace std;
const int mod=7;
struct node{
    int num[2][2];
};
node muil(node A,node B)
{
	node ans;
	memset(ans.num,0,sizeof(ans.num));
	for(int i=0;i<2;i++)
	{
		for(int j=0;j<2;j++)
		{
			for(int k=0;k<2;k++)
			{
				ans.num[i][j]=(ans.num[i][j]+A.num[i][k]*B.num[k][j])%mod;
			}
		}
	}
	return ans;
}
node quickpow(node A,int n)
{
	node ans;
	ans.num[1][1]=1;ans.num[0][0]=1;
	ans.num[1][0]=0;ans.num[0][1]=0;
	while(n)
	{
		if(n%2)
		{
			ans=muil(ans,A);
			//n--;
		}
		A=muil(A,A);
		n>>=1;
	}
	return ans;
}
int main()
{
    int c,a,b,n;
    cin>>a>>b>>n;
    if(n==1||n==2)
    {
        puts("1");
        return 0;
    }
    node A;node ans;
    A.num[0][0]=a;
    A.num[0][1]=b;
    A.num[1][0]=1;
    A.num[1][1]=0;
    ans=quickpow(A,n-2);
    c=ans.num[0][0]+ans.num[0][1];
    //cout<<c<<endl;
    while(c<0) c+=mod;
    cout<<c%mod<<endl;
    return 0;
}

........

看别人的代码他们一眼就能看到题的思路有循环节：https://blog.csdn.net/f_zyj/article/details/52104264

#include <iostream>

using namespace std;

int f[300] = {1, 1, 1};   //  根据规律可知，一定存在循环节

int main(int argc, const char * argv[])
{
    int A, B, N;

    while (cin >> A >> B >> N)
    {
        if (N == 1 || N == 2)
        {
            printf("1\n");
            continue;
        }
        int i;
        for (i = 3; i < 300; i++)
        {
            f[i] = ((A * f[i - 1] + B * f[i - 2]) % 7 + 7) % 7;
            if (f[i - 1] == 1 && f[i] == 1)
            {
                break;
            }
        }
        f[0] = f[i - 2];    //  将循环节尾部值赋值给f[0]
        printf("%d\n", f[N % (i - 2)]);
    }

    return 0;
}