描述
Pell数列a1, a2, a3, ...的定义是这样的,a1 = 1, a2 = 2, ... , an = 2 * an − 1 + an - 2 (n > 2)。
给出一个正整数k,要求Pell数列的第k项模上32767是多少。
输入
第1行是测试数据的组数n,后面跟着n行输入。每组测试数据占1行,包括一个正整数k (1 ≤ k < 1000000)。输出
n行,每行输出对应一个输入。输出应是一个非负整数。
1、循环迭代#include
using namespace std;
int main(){
int n,a,b,c,x;
cin>>n;
a=1,b=2;
while(n--){
a=1,b=2;
cin>>x;
if(x==1)printf("1
");
else if(x==2)printf("2
");
else{
for(int i=3;i<=x;i++){
c=(2*b+a)%32767;
a=b;
b=c;
}
printf("%d
",c%32767);
}
}
return 0;
}
2、递归。直接递归会超时,因此设置一个数组,记忆化一下。#include
using namespace std;
int a[1000000+10];
long long fx(int k){
if(a[k])return a[k];
else return a[k]=(fx(k-1)*2+fx(k-2))%32767;
}
int main(){
int n,k;
cin>>n;
a[1]=1,a[2]=2;
while(n--){
cin>>k;
printf("%lld
",fx(k));
}
return 0;
}