总时间限制: 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
思路:计算出第k项,然后模除32767,输出
方法1:使用循环算出第k项
程序(有误):
#include <iostream>
using namespace std;
#define Max 100000
const int d = 32767;
int pell(int);
int main(){
int n;
int count = 0;
cin >> n;//n行数据
int k;
while(count < n){
cin >> k;
cout << pell(k) <<endl;
count ++;
}
return 0;
}
//计算第k项数据
int pell(int k){
int a_1 = 2,a_2 = 1;
int a = -1;
if(k==1){
a = 1;
}else if(k == 2){
a = 2;
}else{
for(int i = 3;i <= k;i++){
a = 2*a_1+a_2;
a_2 = a_1;
a_1 = a;
}
}
return a;
}
结果:不通过。
分析:int类型范围太小,换成long long ,结果仍然不通过。使用几个数据测试输出一下,发现,即使long long还是超出范围了。这可咋办?
百度后,网友给的答案:
for(int i = 3;i <= k;i++){
a = (2*a_1+a_2)%d;
a_2 = a_1;
a_1 = a;
}
这。。。为什么可以先模除,它们的结果一样嘛?
我用5做除数试一下,
嘶~!居然特喵的一样的!!!那么我不经得出以上这样的结论。
实践说明这个结论是对的,那么问题来了,为什么它是对的?模除的运算法则是啥样来着?
经过一番搜索,俺从这篇中我找到了:
(a + b) % p = (a % p + b % p) % p
(a - b) % p = (a % p - b % p) % p
(a * b) % p = (a % p * b % p) % p
(a^b) % p = ((a % p)^b) % p
https://www.cnblogs.com/juruohx/p/7787415.html
然后,我试着证明一哈它,
证明并不严谨,胡乱一通证,也不知道对不对,就这样吧。。。
方法2:使用递归算出第k项
程序(有误):
#include <iostream>
using namespace std;
#define Max 100000
const int d = 32767;
int pell(int);
int main(){
int n;
int count = 0;
cin >> n;//n行数据
int k;
while(count < n){
cin >> k;
cout << pell(k)%d <<endl;
count ++;
}
return 0;
}
//计算第k项数据
int pell(int k){
if(k==1){
return 1;
}else if(k==2){
return 2;
}else{
return 2*pell(k-1)+pell(k-2);
}
return -1;
}
结果:超时。emm。。。
分析:数字太大,递归实践复杂度太大,100就不动了。
改进:使用打表的方式试试。。。
程序(有误):
#include <iostream>
using namespace std;
#define Max 10000000
const int d = 32767;
int pell(int);
int a[Max];
void init(int []);
int main(){
int n;
int count = 0;
cin >> n;//n行数据
int k;
init(a);
while(count < n){
cin >> k;
cout << pell(k)%d <<endl;
count ++;
}
return 0;
}
void init(int a[]){
a[1] = 1%d;
a[2] = 2%d;
for(int i =3;i<Max;i++){
a[i] = -1;
}
}
//计算第k项数据
int pell(int k){
if(a[k]!=-1){
return a[k];
}else{
a[k] = (2*pell(k-1)+pell(k-2))%d;
return a[k];
}
return -1;
}
结果:效率提高了不少,但k=100000时就没有输出了。对,你没有看错就是直接结束了,没有输出。爆内存了?
不清楚,若有大佬瞧见了,希望提示我一下。