HDOJ:1005---Number Sequence

HDOJ:1005—Number Sequence

题目描述

踩坑记录

这题是斐波那契数列的变种，多了两个参数A和B，一拿到这一题，我首先想到的是利用递归函数的方法，代码如下：

//使用递归 内存过大
#include<stdio.h>

int a,b;

int f(int n){
    if(n==1 || n==2){
        return 1;
    }
    return ( (a*f(n-1)+b*f(n-2))%7 );
}

int main(){
    int n=0;
    while( scanf("%d %d %d",&a,&b,&n) && n>0 && (a>=1&&a<=1000) && (b>=1&&b<=1000)){
        printf("%d\n",f(n));
    } 
}

但是提交的时候Memory Limit Exceeded，占用内存过大，不予通过。
于是我使用循环的方法去求解：

#include<stdio.h>

int main(){
    int a,b,n;
    while( scanf("%d %d %d",&a,&b,&n) && n>0 && a!=0 && b!=0){
        int f1=1,f2=1;
        for(int i=3;i<=n;i++){
            int fn;
            fn = (a*f1+b*f2)%7;
            f1 = f2;
            f2 = fn;
        }

        printf("%d\n",f2);
    } 
}

但是出现了Memory Limit Exceeded，时间又超出了T^T。

正确题解

首先观察f(n)=(a*f(n-1)+b*f(n-2))%7，其f(n)除了n=1，2的时候，其实只有七种取值(n=3,4,5,6,7,8,9)，其他n值的情况都可以通过这个公式，并且利用这七种取值去求，然后有%7，又有七种情况，所以总共是7x7=49种情况。所以可以用数组去存储这49种情况。

AC代码如下：

#include <stdio.h>

int main(){
    int a,b,n,result;
    int f[49]={1,1};     //开头两个元素等于1
    while( scanf("%d %d %d",&a,&b,&n) && n>0 && a!=0 && b!=0){
        for(int i=2;i<49;i++){
            f[i]=(a*f[i-1]+b*f[i-2])%7;
        }
        result=f[n%49-1];
        printf("%d\n",result);
    } 
}