POJ-3070 Fibonacci 矩阵快速幂

上次周赛遇到一道DP+快速幂的题目，结果不会用快速幂，学习一下。

题目给出了公式，求第n位的数，不过只保留后4位，mod 10000 就可以了。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <cmath>
using namespace std;
struct mat
{
	__int64 a[2][2];
};
mat solve(mat p, mat q)
{
	mat c;
	for(int i=0;i<2;i++)
	{
		for(int j=0;j<2;j++)
		{
			c.a[i][j]=0;
			for(int k=0;k<2;k++)
			{
				c.a[i][j]=(c.a[i][j]+p.a[i][k]*q.a[k][j])%10000;
			}
		}
	}
	return c;
}
int main()
{
	int n;
	while(scanf("%d",&n)!=EOF&&n!=-1)
	{
		if(n==0)
		{
			cout<<"0"<<endl;
		}
		else if(n==1)
		{
			cout<<"1"<<endl;
		}
		else
		{
			mat aa,bb;
        	aa.a[0][0]=1;aa.a[0][1]=1;aa.a[1][0]=1;
        	aa.a[1][1]=0;
        	bb.a[0][0]=1;bb.a[1][1]=1;
        	bb.a[0][1]=0;bb.a[1][0]=0;
        	while(n)
        	{
	        	if(n&1)
	        	{
	        		bb=solve(aa,bb);
	        	}
	        	n=n>>1;
	        	aa=solve(aa,aa);
	        }
	        int ans=bb.a[1][0];
	        cout<<ans<<endl;
		}
	}
	return 0;
}