【Pell数列】递归，超时优化

描述

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

经典递归

代码1：【递归超时——当k足够大时，调用pell函数层数太多】

#include <iostream>
#include <cmath>
using namespace std;

int p[1000001];
int pell(int k)
{
    if( k<=2 )
        return k;
    else
        p[k] = 2 * pell(k-1) + pell(k-2);
    
    return p[k]%32767;
}

int main()
{
    int n;
    cin >> n;
    int a[n];
    for(int i=0; i<n; i++)
        cin >> a[i];
    for(int i=0; i<n; i++)
        cout << pell(a[i]) << endl;
    
    return 0;
}

代码2：【用循环】

#include <iostream>
#include <cmath>
using namespace std;

int pell(int k)
{
    if( k<=2 )
        return k;
    else
    {
        int a1=1, a2=2, a3;
        for(int i=0; i<(k-2); i++)
        {
            a3 = (2*a2 + a1)%32767;    //取余操作不会影响值的计算，可以验证
            a1 = a2;
            a2 = a3;
        }
        return a2;
    }
}

int main()
{
    int n;
    cin >> n;
    int a[n];
    for(int i=0; i<n; i++)
        cin >> a[i];
    for(int i=0; i<n; i++)
        cout << pell(a[i]) << endl;
    
    return 0;
}