OpenJudge_P1788 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<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
#define MOD 32767
int a,b,c,t,n;
int main(){
    scanf("%d",&t);
    while(t--){
        scanf("%d",&n);
        if(n<3){
            if(n==1) printf("1\n");
            else printf("2\n");
            continue;
        }
        a=0,b=1,c=2,n-=2;
        while(n--){
            swap(a,b);swap(b,c);c=(a+b*2)%MOD;
        }
        printf("%d\n",c);
    }
}