OJ 1170 数列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

这题呢是一个寻找周期的题目，随着数据逐渐增多最后会出现循环的情况，我们需要找出周期，当然如果数据较小也肯找不到周期，找到周期的条件是f[j-1]==f[i-1]&&f[j]==f[i]，根据此规律即可找出答案了。
 

#include <iostream>
#include <cstring>
using namespace std;
int f[100000];
int main()
{
    int a,b,n,t,p,flag;
    while(cin>>a>>b>>n&&(a||b||n))
    {
        flag=0;
        f[1]=f[2]=1;
        for(int i=3;i<=n;i++)
        {
            f[i]=(a*f[i-1]+b*f[i-2])%7;
            for(int j=2;j<i;j++)
            {
                 if(f[j-1]==f[i-1]&&f[j]==f[i])
                {
                    flag=1;
                    t=j;
                    p=i;
                    break;
                }

            }
            if(flag==1)
                break;
        }
        if(flag==1)
        cout<<f[t+(n-p)%(t-p)]<<endl;
        else
        cout<<f[n]<<endl;
    }
    return 0;
}