HDU-1005-Number Sequence

时间限制:2000/1000 MS(Java /其他)内存限制:65536/32768 K(Java /其他)
提交总计:239120接受提交:60940

问题描述
一个数字序列定义如下:

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 <= 100,000,000)。三个零表示输入结束,该测试用例将不被处理。

输出量
对于每个测试用例,将f(n)的值打印在一行上。

样本输入
1 1 3
1 2 10
0 0 0

样本输出
2

对递推公式用递归会爆栈,一定要优化以下,我是用数组优化的,然后这里mod7,所以f【i】
一定是在0,1,2,3,4,5,6之间的,所以对于任意一个元素f【i】有7种值,乘上不同的系数A和B之后,Af【i-1】和Bf【i-2】分别都有7种不同的值,作和之后f【i】会有7个7种,就是49种值,之后就是循环了

#include <iostream>

using namespace std;

int main()
{
    int i=0,n=0,A=0,B=0,a[10000];
    a[1]=a[2]=1;
    while(cin>>A>>B>>n)
    {

        if(A==0&&B==0&&n==0)
            break;
        for(i = 3 ; i <= 48 ; i++)
        {
            a[i]=(A*a[i-1]+B*a[i-2])%7;
        }
        cout << a[n%49] << endl;
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_46028606/article/details/106583710