涂抹果酱

涂抹果酱（状压）

题面 ：yvj 两周年庆典要到了，Sam 想为 Tyvj 做一个大蛋糕。蛋糕俯视图是一个 N×M 的矩形，它被划分成 N×M 个边长为 1×1 的小正方形区域（可以把蛋糕当成 N 行 M 列的矩阵）。蛋糕很快做好了，但光秃秃的蛋糕肯定不好看！所以，Sam 要在蛋糕的上表面涂抹果酱。果酱有三种，分别是红果酱、绿果酱、蓝果酱，三种果酱的编号分别为 1,2,3。为了保证蛋糕的视觉效果，Admin 下达了死命令：相邻的区域严禁使用同种果酱。但 Sam 在接到这条命令之前，已经涂好了蛋糕第 K 行的果酱，且无法修改。

现在 Sam 想知道：能令 Admin 满意的涂果酱方案有多少种。请输出方案数 mod 10^6。若不存在满足条件的方案，请输出 0。

这道题显然是状压dp 但是是三进制的，我们需要用一个数组（db []）来记录3的k次方db[i]=db[i-1]*3; 同时 可以提前找出合法状态

#include<bits/stdc++.h>
using namespace std;
const int N=11000;
const int M=10;
int n,m,dp[N][250],k,mp[1000],ok[250][250];
int db[8];
#define mod 1000000
inline int check1(int x){
	for(int i=m-1;i;i--){
		if((x%db[i+1])/db[i]==(x%db[i])/db[i-1])return 0; 
	}
	return 1;
}
inline int check2(int x,int y){
	for(int i=m-1;i>=0;i--){
		if((x%db[i+1])/db[i]==(y%db[i+1])/db[i])return 0; 
	}
	return 1;
}
int main(){
	cin>>n>>m>>k;
	int sum=0;
	db[0]=1;
	for(int i=1;i<=m;i++){
		int x;cin>>x;
		sum=sum*3+(x-1);
		db[i]=db[i-1]*3;
	}
	if(!check1(sum)){
		puts("0");return 0;
 	}
 	int t=0;
 	for(int i=0;i<db[m];i++){
 		if(check1(i)) mp[++t]=i;
 		if(sum==i) dp[0][t]=1;
  	} 
	 for(int i=1;i<=t;i++)
	    for(int j=1;j<=t;j++){
	    	ok[i][j]=check2(mp[i],mp[j]);
	    }	
  	int x=min(n-k,k-1),y=max(n-k,k-1);
	  for(int i=1;i<=y;i++)
	      for(int j=1;j<=t;j++)
	           for(int k=1;k<=t;k++){
	               if(ok[k][j]) {
	               	(dp[i][j]+=dp[i-1][k])%=mod;
	            }
	}
	int ans1=0;int ans2=0;
	for(int i=1;i<=t;i++){
	ans1=(ans1+dp[x][i])%mod;ans2=(ans2+dp[y][i])%mod;
	}
	printf("%lld",((long long)ans1*ans2)%mod);
	return 0;
}