Blocks POJ 3734

判断好三种方块会出现的条件后就可以解出，算法是矩阵快速幂的模版

#include<iostream> 
#include<cstring>
#include<algorithm>

using namespace std;

const int MOD=10007;

struct node{
	int a[5][5];
	node operator * (const node &b){
		node ans;
		memset(ans.a,0,sizeof(ans.a));
		for(int i=0;i<4;i++)
			for(int j=0;j<4;j++)
				for(int k=0;k<4;k++)
				{
					ans.a[i][j]+=a[i][k]*b.a[k][j]%MOD;
					ans.a[i][j]%=MOD;
				}
		return ans;
	}
};

node powk(node a,int b){
	node pos;
	memset(pos.a,0,sizeof(pos.a));
	for(int i=0;i<4;i++)
		pos.a[i][i]=1;
	while(b)
	{
		if(b&1)
			pos=pos*a;
		b/=2;
		a=a*a;
	}
	return pos;
}

int main()
{
	int t;
	cin>>t;	
	while(t--)
	{
		int n;
		node ans;
		cin>>n;
		int a[4][4]={{2, 1, 1, 0},{1, 2, 0, 1},{1, 0, 2, 1},{0, 1, 1, 2}};
		for(int i=0;i<4;i++)
			for(int j=0;j<4;j++)
				ans.a[i][j]=a[i][j];
		ans=powk(ans,n);
		cout<<ans.a[0][0]%MOD<<endl;
	}
	return 0;
}