菲波那契数列是指这样的数列: 数列的第一个是0和第二个数是1,接下来每个数都等于前面2个数之和。 给出一个正整数a,要求菲波那契数列中第a个数的后四位是多少。
Input
多组数据 -1结束 范围1~10^9
Output
第x项的后4位
Sample Input
0 9 999999999 1000000000 -1
Sample Output
0 34 626 6875
前言:因为对矩阵快速幂模板不太熟悉,就以这个题为例,记录一下矩阵快速幂的模板吧。
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3f;
const int mod=1e4;
const int f=2;//这个值是你需要的常数矩阵和第一项矩阵的大小
struct node
{
ll matrix[5][5];//定义一个结构体用来存矩阵,比你需要的矩阵大一些
};
node mul(node a,node b)//矩阵乘法
{
node ans;
memset(ans.matrix,0,sizeof(ans.matrix));
for(int i=1;i<=f;i++)
for(int j=1;j<=f;j++)
for(int k=1;k<=f;k++)
ans.matrix[i][j]=(ans.matrix[i][j]+a.matrix[i][k]*b.matrix[k][j]%mod+mod)%mod;
return ans;
}
node ksm(node a,int b)//矩阵乘法
{
node res;
memset(res.matrix,0,sizeof(res.matrix));
for(int i=1;i<=f;i++) res.matrix[i][i]=1;
while(b)
{
if(b&1) res=mul(res,a);
a=mul(a,a);
b>>=1;
}
return res;
}
int main()
{
// freopen("input.txt","r",stdin); //提交时关闭
std::ios::sync_with_stdio(false);
int n;
while(cin>>n&&n!=-1)
{
if(n==0||n==1)
{
cout<<n<<endl;
continue;
}
node a,b;//a为第一项矩阵,b为常数矩阵
memset(a.matrix,0,sizeof(a.matrix));
memset(b.matrix,0,sizeof(b.matrix));
a.matrix[1][1]=1,a.matrix[1][2]=0;
a.matrix[2][1]=0,a.matrix[2][2]=0;
b.matrix[1][1]=1,b.matrix[1][2]=1;
b.matrix[2][1]=1,b.matrix[2][2]=0;
node ans;
memset(ans.matrix,0,sizeof(ans.matrix));
ans=ksm(b,n-1);//先让常数矩阵快速幂n-1次
ans=mul(ans,a);//再乘以第一项矩阵
cout<<ans.matrix[1][1]<<endl;//输出所求第一项
}
return 0;
}