数列2（循环节）

数列2
描述

小明今天在做数学题的时候碰到这样一个问题，一个数列的定义如下：
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（1<=A,B<=1000，1<=n<=100000000），当输入的3个数都为0时，输入结束。

输出
对于每组输入，输出f(n)的值。

输入样例 1

1 1 3
1 2 10
0 0 0
输出样例 1

2
5
来源

天勤考研算法练习赛题目-天勤考研算法练习赛（4）

这题我也是百度看了别人怎么做，第一次做题有循环节的概念，取模运算一般都会出现循环节，具体可以百度下。

这题得注意下：

1.f[n]可能没有到经过一个循环，这个数一般比较小，直接输出答案。

2.判断循环节，这里采用只判断循环的前两个数f[i-1]==f[j-1]&&f[i]==f[j]

3.循环节的长度可能为1，也就是所有的元素都相等。

#include<bits/stdc++.h>
using namespace std;
int f[10000];
int main()
{
    int a,b,n;
    while(cin>>a>>b>>n)
    {
        int i,j,s=0,e=0,flag=0;
        if(a==0&&b==0&&n==0)break;
        f[1]=f[2]=1;
        for(i=3;i<=n;i++)
        {
            f[i]=(a*f[i-1]+b*f[i-2])%7;
            flag=0;
            for(j=2;j<i;j++)
            {
                if(f[j-1]==f[i-1]&&f[j]==f[i])
                {
                    flag=1;
                    s=j;
                    e=i;
                    break;
                }
            }
            if(flag==1)break;
        }
        if(flag)   //如果有循环节
            cout<<f[s+(n-e)%(e-s)]<<endl;
        else    //没有循环节
            cout<<f[n]<<endl;
    }
    return 0;
}