Pell数列

总时间限制: 3000ms 内存限制: 65536kB
描述
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行，每行输出对应一个输入。输出应是一个非负整数。
样例输入

2
1
8

样例输出

1
408

代码

#include<iostream>
#include<cstdio>
using namespace std;
int main(){
	int n,a1=1,a2=2,a3,k=0;
	long long tmp,a[1005];
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>tmp;
		a1=1;
		a2=2;
		if(tmp==1){
			a[k++]=a1;
		}
		else if(tmp==2){
			a[k++]=a2;
		}
		else{
			for(int j=3;j<=tmp;j++){
				a3=(2*a2+a1)%32767;
			    a1=a2;
			    a2=a3;
			}
			a[k++]=a3;
		}
	}
	for (int i=0; i<k; i++) {
		cout<<a[i]<<endl;
    }
	return 0;
}