[NOI题库][POJ2786]Pell数列

总时间限制: 3000ms 内存限制: 65536kB

描述

Pell数列a₁, a₂, a₃, ...的定义是这样的，a₁ = 1, a₂ = 2, ... , a_n = 2 * a_{n ? 1} + a_{n - 2} (n > 2)。
给出一个正整数k，要求Pell数列的第k项模上32767是多少。

输入

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

输出

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

样例输入

2
1
8

样例输出

1
408

代码

#include <stdio.h>
#include <string.h>

int a[ 1000010 ];
int n, k;
const int mo = 32767;

int main(){
    scanf("%d", &n);

    int last = 3;
    a[ 1 ] = 1;
    a[ 2 ] = 2;
    int i, j;
    for( i = 1; i <= n; i++ ){
        scanf("%d", &k);

        for( j = last; j <= k; j++ ){
            a[ j ] = (((2 * a[ j - 1 ]) % mo) + a[ j - 2 ]) % mo;
        }
        if( k > last ) last = k;
        printf("%d\n", a[ k ] % mo);
    }

    return 0;
}