Fibonacci

题目链接

菲波那契数列是指这样的数列: 数列的第一个是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;
}