poj3734 Blocks 矩阵的幂

http://poj.org/problem?id=3734

#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const long long MOD=10007;//int报错
struct mat{
	long long a[3][3];
};
mat mat_mul(mat x,mat y){
	mat res;
	memset(res.a,0,sizeof(res.a));
	for(int i=0;i<3;i++){
		for(int j=0;j<3;j++){
			for(int k=0;k<3;k++){
				res.a[i][j]=(res.a[i][j]+x.a[i][k]*y.a[k][j])%MOD;
			}
		}
	}
	return res;
}
mat mat_pow(int n){
	mat c,res;
	memset(res.a,0,sizeof(res.a));
	c.a[0][0]=c.a[1][0]=c.a[1][1]=c.a[1][2]=c.a[2][2]=2;
	c.a[0][1]=c.a[2][1]=1;
	c.a[0][2]=c.a[2][0]=0;
	for(int i=0;i<3;i++){
		res.a[i][i]=1;
	}
	while(n){
		if(n&1){
			res=mat_mul(res,c);
		}
		c=mat_mul(c,c);
		n=n>>1; 
	}
	cout<<res.a[0][0]<<endl;
}
int main(){
	int T,n;
	cin>>T;
	while(T--){
		cin>>n;
		mat_pow(n);
	}
	return 0;
}