矩阵快速幂求斐波那契数列

#include <iostream>
using namespace std;
const int MOD=10000;
struct M{
	int m[2][2];
};
M mul(M a,M b){   //矩阵乘法,计算矩阵a×矩阵b
	M ans;
	for(int i=0;i<2;i++)
		for(int j=0;j<2;j++){
			ans.m[i][j]=0;
			for(int k=0;k<2;k++)
				ans.m[i][j]=(ans.m[i][j]+a.m[i][k]*b.m[k][j])%MOD;
		}
	return ans;
}
M fast(M a,int n){  //矩阵快速幂，计算矩阵a的n次幂
	M ans;
	M aa=a;
	ans.m[0][0]=ans.m[1][1]=1;
	ans.m[0][1]=ans.m[1][0]=0;
	while(n){
		if(n&1) ans=mul(ans,aa);
		aa=mul(aa,aa);
		n>>=1;
	}
	return ans;
}
int cal(int n){    //利用递推公式计算斐波那契数列第n项
	if(n==0) return 0;
	if(n==1) return 1;
	M tmp;
	tmp.m[0][0]=tmp.m[0][1]=tmp.m[1][0]=1;    //广义斐波那契数列递推公式不同，则矩阵tmp元素可变
	tmp.m[1][1]=0;
	M ans=fast(tmp,n-1);
	return ans.m[0][0];
}
int main(){
	int n;
	while(cin>>n)
		cout<<cal(n)<<endl;
	return 0;
}