NoiOpenJudge 2.2 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 <iostream>
using namespace std;
int main()
{
    long k,a,b,c;
    int i,n,j;
    cin>>n;
    for(i=1;i<=n;i++)
    {
        cin>>k;
        a=1;
        b=2;
        if(k==1)cout<<a<<endl;
        else if(k==2)cout<<b<<endl;
        else
        {
            for(j=3;j<=k;j++)
            {
                c=(2*b+a)%32767;
                a=b;
                b=c;
            }
            cout<<c<<endl;
        }
    }
}