HOJ 2662 Pieces Assignment（状态压缩DP）

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;

long long dp[82][22][1<<9],ans;//dp表示第i行用j个棋子状态x下的方法数; 
int mark[1<<9],len;//对于单独某一行来说，合法结果是一定的。将所有的这些合法结果保存 


int m,n,k;//m表示行，n表示列 
long long count(long long x) //记录状态x中1的个数  
{  
    long long sum=0;  
    while(x)  
    {  
        if(x&1)  
            sum++;  
        x=x>>1;  
    }  
    return sum;  
}  
  
bool judge(long long x) //判断状态x是否有相邻的棋子放在一起  
{  
  
    if(x&(x<<1)) return false;  
    return true;  
}  


int main()
{
	
	while(scanf("%d%d%d",&m,&n,&k)!=EOF)
	{
		if(m>n)
		   swap(m,n);//让大的数为行,小的数为列.由于行与列居然对称性，这样的操作不影响结果 

		ans=0;
		len=0;   
		memset(dp,0,sizeof(dp));
		for(int i=0;i<(1<<m);i++)//在对第一个行初始化处理的同时，对合法的状态进行保存 
		{
		 if(judge(i))
		 {
		 	dp[1][count(i)][len]=1;//第三维不直接写i的原因是为了方便遍历 
		 	mark[len++]=i;//一个len对于一个状态的十进制数 
		 }	
		}
		
		for(int i=2;i<=n;i++)//从第2行到第n行 
		   for(int j=0;j<=k;j++)//当前行用0到k个棋子 
		      for(int x=0;x<len;x++)//枚举第i行的状态 
			  {
			  	for(int y=0;y<len;y++)//枚举i-1行的状态 
			  	{
			  		int temp=count(mark[x]);
			  		if(((mark[x]&mark[y])==0)&&j>=temp)//一定要注意优先级 
			  		dp[i][j][x]+=dp[i-1][j-temp][y];		
			  	}	
			  }    
		
		for(int i=0;i<len;i++)
		  ans+=dp[n][k][i];
		
		printf("%lld\n",ans);
	
	}
	
	return 0;
}