两道题基本差不多,还是这个套路写,比之前的取数多了一个是否适应题目给定状态的判断,大题思路是雷同的。POJ 3254那个题目像是这两个题目的基础。先A了那个题目,这两道做起来顺手多了。
这两个题目都不仅牵扯到上一行,还牵扯到上上一行,用了三维数组记录,dp[i][j][k]代表在第i行,本行状态是j上一行状态是k的时候能取到的最大值。
附上HDU4539的代码:
#include<stdio.h>
#include<string.h>
int num[125],dp[125][200][200];
int s[200];
int max(int a,int b)
{
if(a>b)
return a;
return b;
}
void inti()//预处理
{
int i,k=0;
for(i=0;i<=2048;i++)
{
if(i&(i<<2))
continue;
s[k++]=i;
}
}
int fit(int i,int x)//判断是否符合硬性条件
{
if(x&num[i])
return 0;
return 1;
}
int getsum(int x)//求出x状态多少个1
{
int ans=0;
while(x)
{
if(x&1)
ans++;
x=x/2;
}
return ans;
}
int main()
{
int n,m,i,j,k;
inti();
while(scanf("%d %d",&n,&m)!=EOF)
{
int x;
for(i=0;i<n;i++)
{
num[i]=0;
for(j=0;j<m;j++)
{
scanf("%d",&x);
if(x==0)
num[i]+=1<<j;
}
}
x=1<<m;
memset(dp,-1,sizeof(dp));
for(i=0;s[i]<x;i++)//首行单独处理
{
if(!fit(0,s[i]))
continue;
dp[0][i][0]=getsum(s[i]);
}
for(i=1;i<n;i++)
{
for(j=0;s[j]<x;j++)
{
if(!fit(i,s[j]))
continue;
int sum=getsum(s[j]);
for(k=0;s[k]<x;k++)
{
if(s[j]&(s[k]<<1)||s[j]&(s[k]>>1))
continue;
for(int l=0;s[l]<x;l++)
{
if(s[j]&s[l])
continue;
if(s[k]&(s[l]<<1)||s[k]&(s[l]>>1))//这个不是太有必要- -,只是当时数组开小不知道错在哪加上去了
continue;
if(dp[i-1][k][l]==-1)
continue;
dp[i][j][k]=max(dp[i][j][k],dp[i-1][k][l]+sum);
}
}
}
}
int ans=0;
for(j=0;s[j]<x;j++)
{
for(k=0;s[k]<x;k++)
ans=max(ans,dp[n-1][j][k]);
}
printf("%d\n",ans);
}
return 0;
}