Pell数列（递推）

描述

Pell数列a1,a2,a3,...的定义是这样的，a1=1,a2=2,...,an=2an−1+an−2(n > 2).

给出一个正整数k，要求Pell数列的第k项模上32767是多少。

格式

输入格式

第1行是测试数据的组数n(n不超过1000)，后面跟着n行输入。每组测试数据占1行，包括一个正整数k (1≤k<1000000)。

输出格式

n行，每行输出对应一个输入。输出应是一个非负整数。

样例

输入样例

2
1
8

输出样例

扫描二维码关注公众号，回复： 10206284 查看本文章

1
408

限制

时间限制: 10000 ms

内存限制: 65536 KB

 注意点1：“第k项模上32767”，每次计算都要模32767

注意点2：每次开始循环，都得设a1=1，a2=2

#include <iostream>
#include <math.h>
#include <string.h>
#include <algorithm>
using namespace std;

int main ()
{
	int n, a1=1, a2=2, a3, k=0;
	long long tmp, a[1005];
	scanf ("%d", &n);
	for (int i=0; i<n; i++) {
		scanf ("%d", &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++) {
		printf ("%d\n", a[i]);
	}
	return 0;
}