跳舞毯

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2543    Accepted Submission(s): 1206

由于长期缺乏运动，小黑发现自己的身材臃肿了许多，于是他想健身，更准确地说是减肥。
小黑买来一块圆形的毯子，把它们分成三等分，分别标上A,B,C，称之为“跳舞毯”，他的运动方式是每次都从A开始跳，每次都可以任意跳到其他块，但最后必须跳回A，且不能原地跳.为达到减肥效果，小黑每天都会坚持跳n次，有天他突然想知道当他跳n次时共几种跳法，结果想了好几天没想出来－_－
现在就请你帮帮他，算出总共有多少跳法。

2
3
4
0

2
2
6

这题数据太小了可以用暴力

我这里用矩阵快速幂做

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<map>
#include<iomanip>
#define INF 99999999
using namespace std;

const int MAX=3;
const int mod=10000;
int array[MAX][MAX],sum[MAX][MAX];

void MatrixMult(int a[3][3],int b[3][3]){
	int c[3][3]={0};
	for(int i=0;i<3;++i){
		for(int j=0;j<3;++j){
			for(int k=0;k<3;++k){
				c[i][j]+=a[i][k]*b[k][j];
			}
		}
	}
	for(int i=0;i<3;++i){
		for(int j=0;j<3;++j)a[i][j]=c[i][j]%mod;
	}
}

int Matrix(int k){
	array[0][0]=array[1][1]=array[2][2]=0;
	array[0][1]=array[0][2]=array[1][0]=array[1][2]=1;
	array[2][0]=array[2][1]=1;
	memset(sum,0,sizeof sum);
	sum[0][0]=sum[1][1]=sum[2][2]=1;
	while(k){
		if(k&1)MatrixMult(sum,array);
		MatrixMult(array,array);
		k>>=1;
	}
	return sum[0][0];
}

int main(){
	int n;
	while(scanf("%d",&n),n){
		printf("%d\n",Matrix(n));
	}
	return 0;
}