POJ-1185-炮兵阵地

因为腾讯马拉松的关系,第一次系统学习状态压缩吧,建议大家可以看看天津大学周伟的PPT,这个题也是上面的一个例题。

代码:

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
const int maxn=101;
int dp[maxn][maxn][maxn],s[maxn],c[maxn],e[maxn];
int n,m,cnt;
int Count(int x)
{
    int ans=0;
    while(x)
    {
	if(x&1)
	    ans++;
	x>>=1;
    }
    return ans;
}
int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
	char str[11];
	memset(e,0,sizeof(e));
	for(int i=0;i<n;i++)
	{
	    scanf("%s",str);
	    for(int j=0;j<strlen(str);j++)
		e[i]=str[j]=='H'?(e[i]<<1)|1:e[i]<<1;
	}
	cnt=0;
	int total=1<<m;
	for(int i=0;i<total;i++)
	{
	    if(i&(i<<1)||i&(i<<2))
		continue;
	    s[cnt]=i;
	    c[cnt++]=Count(i);
	}
	memset(dp,-1,sizeof(dp));
	for(int i=0;i<cnt;i++)
	{
	    if(s[i]&e[0])
		continue;
	    dp[0][i][0]=c[i];
	}
	for(int i=1;i<n;i++)
	{
	    for(int j=0;j<cnt;j++)
	    {
		if(e[i]&s[j])
		    continue;
		for(int k=0;k<cnt;k++)
		{
		    if(s[j]&s[k])
			continue;
		    for(int l=0;l<cnt;l++)
		    {
			if(s[j]&s[l])
			    continue;
			if(dp[i-1][k][l]==-1)
			    continue;
			dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+c[j]);
		    }
		}
	    }
	}
	int ans=0;
	for(int i=0;i<cnt;i++)
	    for(int j=0;j<cnt;j++)
		ans=max(ans,dp[n-1][i][j]);
	printf("%d\n",ans);
	
    }
    return 0;
}


你可能感兴趣的:(dp,状态压缩)