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);
}
}