状压的第一题:
#include<cstdio>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<map>
#include<cmath>
#include<iostream>
#include <queue>
#include <stack>
#include<algorithm>
#include<set>
using namespace std;
#define INF 1e8
#define inf -0x3f3f3f3f
#define eps 1e-8
#define LL long long
#define M 100001
#define mol 100000007
bool lp(int a,int b)//判断两个状态共存时是否合法,合法为0,不合法为非0
{
return a&b;
}
int dp[105][100][100];//dp[i][j][k] 表示第i行状态为state[j],第i-1行状态为state[k]时的最优解
int base[105];//第i行的原地图压缩成的一个状态
int state[105];//仅是两个炮兵不互相攻击的条件下,符合条件的状态(一个十进制数)
int soldier[105];//对应着,在state[i]状态下能放多少个士兵
int num;//仅是两个炮兵不互相攻击的条件下,符合条件的状态个数
int main()
{
int n,m;
int i,j,k;
char a[105][15];
while(cin>>n>>m)
{
memset(base,0,sizeof(base));
memset(dp,0,sizeof(dp));
memset(state,0,sizeof(state));
memset(soldier,0,sizeof(soldier));
for(i=0;i<n;i++)
{
cin>>a[i];
for(j=0;j<m;j++)
{
if(a[i][j]=='H')
base[i]+=1<<j;//先计算原始地图的状态数
}
}
num=0;
for(i=0;i<(1<<m);i++)//仅是两个炮兵不互相攻击的条件下计算所有状态
{
if(lp(i,i<<1)||lp(i,i<<2)) continue;
k=i;
while(k)
{
soldier[num]+=k&1;//这个循环计算状态i的二进制形式里面有多少个1,也就是放了多少个士兵
k=k>>1;
}
state[num++]=i;//保存这个合法的状态
}
for(i=0;i<num;i++)//先初始化dp[0][i][0],即初始化第1行的情况
{
if(lp(state[i],base[0])) continue;//在state[i]的基础上,还要满足士兵不能放在山上,这个判断就是处理这个问题的
dp[0][i][0]=soldier[i];
}
for(i=0;i<num;i++)//接着初始化dp[1][i][j],即第2行的情况
{
if(lp(state[i],base[1])) continue;
for(j=0;j<num;j++)//枚举第1行的状态
{
if(lp(state[j],base[0])) continue;
if(lp(state[j],state[i])) continue;
dp[1][i][j]=max(dp[1][i][j],dp[0][j][0]+soldier[i]);
}
}
for(int r=2;r<n;r++)//第3行开始DP直到最后
{
for(i=0;i<num;i++)
{
if(lp(state[i],base[r])) continue;
for(j=0;j<num;j++)//枚举第r-1行的状态
{
if(lp(state[j],base[r-1])) continue;
if(lp(state[i],state[j])) continue;
for(k=0;k<num;k++)//枚举第r-2行的状态
{
if(lp(state[k],base[r-2])) continue;
if(lp(state[k],state[i])) continue;
if(lp(state[i],state[k])) continue;
dp[r][i][j]=max(dp[r][i][j],dp[r-1][j][k]+soldier[i]);
}
}
}
}
int ans=0;
for(i=0;i<num;i++)
for(j=0;j<num;j++)
ans=max(ans,dp[n-1][i][j]);
cout<<ans<<endl;
}
return 0;
}