NOIP学习之递归:187.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
#define k 32767
#define Max 1000000
using namespace std;
int ans[Max+5];
int math(int m)
{
	if(m<3) return m;
	if(ans[m]!=0) return ans[m];
	if(ans[m]==0)
		return ans[m]=(2*math(m-1)%k+math(m-2)%k)%k;
}
int main(){
	int n,a;
	cin>>n;
	for(int i=0;i<n;i++)
	{
		cin>>a;
		cout<<math(a)<<endl;		
	}
	return 0;
}