Pell数列

问题描述

Pell数列a1, a2, a3, ...的定义是这样的，a1 = 1, a2 = 2, ... , an = 2 * an − 1 + an - 2 (n > 2)。
给出一个正整数k，要求Pell数列的第k项模上32767是多少。

输入格式

第1行是测试数据的组数t，表示接下来有t组测试数据(1<=t<=100)。
每组测试数据占1行，包括一个正整数k (1 ≤ k ≤ 20000)。

输出格式

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

样例输入

3
1
8
20

样例输出

1
408
4132

#include <iostream>
using namespace std;
int pell(int k) {
    if (k == 1) {  // 第一项为1
        return 1;
    }
    if (k == 2) {  // 第二项为2
        return 2;
    }
    int a1 = 1, a2 = 2, ak;
    for (int i = 3; i <= k; i++) {
        ak = (2 * a2 + a1) % 32767;  // 计算第i项
        a1 = a2;  // 更新a1和a2
        a2 = ak;
    }
    return ak;
}
int main() {
    int k;
    cin >> k;
    cout << pell(k) << endl;
    return 0;
}