题目描述
NowCoder最近在研究一个数列：
- F(0) = 7
- F(1) = 11
- F(n) = F(n-1) + F(n-2) (n≥2)
他称之为NowCoder数列。请你帮忙确认一下数列中第n个数是否是3的倍数。
输入描述：
输入包含多组数据。
每组数据包含一个整数n，(0≤n≤1000000)。
输出描述
对应每一组输入有一行输出。
如果F(n)是3的倍数，则输出“Yes”；否则输出“No”。
输入例子：
0
1
2
3
4
5
输出例子：
No
No
Yes
No
No
No
题目分析
这是一个特殊初始条件的Fibonacci sequence.
1.初始条件n的值比较小，可以直接枚举每个F(n)的值，然后输入n查询：
#include <iostream>
#include <cstdio>
using namespace std;
int main (){
int n;
int *a=new int[N];
a[0]=7;
a[1]=11;
for(int i=2;i<N;i++)
a[i]=a[i-2]%3+a[i-1]%3;
while(~scanf("%d", &n)){
cout<<a[n]<<endl;
if(a[n]%3==0)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
return 0;
}
2.可以寻找到规律：（牛客网友“我要过pat”提供）
这个真牛批，想不到还有这种规律。那Fibonacci数列是否有相似的规律呢？
#include <iostream>
using namespace std;
int main (){
int n;
while(cin>>n){
if((n-2)%4==0)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
return 0;
}
3.用矩阵快速幂来求解，这题n的范围比较小(0≤n≤1000000)所以用第1种方法可以不超时求解，但是，倘若n很大，达到(1<=n<=1000,000,000,000,000,000)这个范围，那么显而易见，必定超时。而且根本开不到那么大的数组。学校的算法课，有另外一题类似的，可以思考一下：
借此思路，用矩阵快速幂计算，稍微修改下就可以了。
#include <iostream>
#include <cstdio>
#include <cstring>
#define N 3
#define maxn 2
using namespace std;
struct Matrix{
long long a[maxn][maxn];
void init(){ //初始化为单位矩阵
memset(a, 0, sizeof(a));
for(int i=0;i<maxn;++i){
a[i][i] = 1;
}
}
};
//矩阵乘法
Matrix mul(Matrix a, Matrix b){
Matrix ans;
for(int i=0;i<maxn;++i){
for(int j=0;j<maxn;++j){
ans.a[i][j] = 0;
for(int k=0;k<maxn;++k){
ans.a[i][j] += a.a[i][k] * b.a[k][j];
ans.a[i][j] %= N;
}
}
}
return ans;
}
//矩阵快速幂
Matrix qpow(Matrix a, long long n){
Matrix ans;
ans.init();
while(n){
if(n&1)
ans = mul(ans, a);
a = mul(a, a);
n /= 2;
}
return ans;
}
int main (){
long long n;
while(~scanf("%lld", &n)){
Matrix a;
a.a[0][0] = 1;
a.a[0][1] = 1;
a.a[1][0] = 1;
a.a[1][1] = 0;
long long s0=7,s1=11;
if (n>1)
{
Matrix ans= qpow(a, n-1);
long long res=ans.a[0][0]*s1+ans.a[0][1]*s0;
// printf("%lld",res);
if(res%3==0)
cout<<"Yes"<<endl;
else
cout<<"No"<<endl;
}
else
cout<<"No"<<endl;
}
return 0;
}
测评网站真的恶心，有时候循环输入用"cin>>"就可以，有时候又超时。。。得改"成~scanf("%lld", &n)"。另外一开始没想到对中间结果对3取模，导致数值过大溢出。用到unsigned long long 都不够。。。
PS：检测一个数被3整除的算法
如果所有的偶数位出现1的次数为 even_count, 奇数位出现1的次数为 odd_count，两者只差如果是3的倍数，那么这个数就是3倍数。
现在给出一个数a，假设它能被3整除，结果是b，即a=3*b，那么从二进制乘法运算判断出，b的最低位与a的最低位一定是相同的，从而得到了b的最低位，将这个位左移1位变成次低位，那么a的次低位以上的比特减去这个位后在次低位上的结果一定是b的次低位。以此类推可以求出b的各个比特，如果最后能完成对b的各位的计算，那么a能够被3整除，否则不能被3整除。
那么算法原理是什么呢？