hdu 1185（状态压缩dp）

点击打开链接

题意：

很之前做的那一道很像，只不过是上下3个不能相邻。。。。

之前Ok函数写错了，RE了好几次才发现。。

#include"stdio.h"
#include"string.h"
#define N 110
int n,m;
int cnt;
int cur[N];
int state[70];
int dp[N][70][70];
int ok(int x)
{
	if(x&(x<<1))return 0;
	if(x&(x<<2))return 0;
	return 1;
}
void init()
{
	int i;
	cnt=0;
	for(i=0;i<(1<<m);i++)
		if(ok(i))state[cnt++]=i;
}
int fit(int x,int k)
{
	if(cur[x]&state[k])return 0;
	return 1;
}
int count(int x)
{
	int ans=0;
	while(x)
	{
		ans+=x%2;
		x/=2;
	}
	return ans;
}
int max(int x,int y)
{
	return x>y?x:y;
}
int main()
{
	int i,j,k,l;
	char s[11];
	int A[N];
	while(scanf("%d%d",&n,&m)!=-1)
	{
		getchar();
		init();
		for(i=0;i<n;i++)
		{
			gets(s);
			cur[i]=0;
			for(j=0;s[j];j++)
			{
				if(s[j]=='H')cur[i]+=(1<<(m-1-j));
			}
		}
		memset(dp,-1,sizeof(dp));
		for(i=0;i<cnt;i++)
		{
			A[i]=count(state[i]);
			if(fit(0,i))dp[0][0][i]=A[i];
		}
		for(i=1;i<n;i++)
		{
			for(j=0;j<cnt;j++)
			{
				if(!fit(i,j))continue;
				for(k=0;k<cnt;k++)
				{
					if(state[j]&state[k])continue;
					for(l=0;l<cnt;l++)
					{
						if(state[j]&state[l])continue;
						if(dp[i-1][l][k]==-1)continue;
						dp[i][k][j]=max(dp[i][k][j],dp[i-1][l][k]+A[j]);
					}
				}
			}
		}
		int ans=0;
		for(i=0;i<n;i++)
		{
			for(j=0;j<cnt;j++)
			{
				for(k=0;k<cnt;k++)
					ans=max(ans,dp[i][j][k]);
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}