题目链接:[NOI2001] 炮兵阵地 - 洛谷
思路:类似“互不侵犯”那道题,只不过影响范围变大了,还有加入的地形影响。所以用了类似的写法,但是不知道为什么wa了,debug了好久也没修好,先放着吧。
#include <bits/stdc++.h>
using namespace std;
int n, m, bit[10], cnt[10], size;
int f[100][1<<10][1<<10]; //行数,上一行,当前行
bool mp[100][10]; int g[100];
inline int read(){
int x=0;char ch=getchar();
while (ch>='0'&&ch<='9'){x=x*10+ch-48;ch=getchar();}
return x;
}
void dfs(int b,int c,int pos){ //预处理一行的情况
if(pos>=m) {bit[++size]=b; cnt[size]=c; return;}
dfs(b,c,pos+1);
dfs(b+(1<<pos),c+1,pos+3);
}
int main(){
ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
n=read(); m=read();
dfs(0,0,0); //预处理
//1.处理地图
char ch;
for(int i=0; i<n; i++)
for(int j=0; j<m; j++){
cin>>ch;
g[i] = (g[i]<<1)+(ch=='H');
}
//补充:n=1的特判
if(n==1){
int ans = 0;
for(int i=1; i<=size; i++){
if(bit[i]&g[0]) continue;
ans = max(ans, cnt[i]);
}
cout<<ans;
return 0;
}
//2.初始化前两行
for(int i=1; i<=size; i++)
for(int j=1; j<=size; j++){
if((g[0]&bit[i]) || (g[1]&bit[j]) || (bit[i]&bit[j])) continue;
f[1][bit[i]][bit[j]] = cnt[i]+cnt[j];
}
//3.状压dp
for(int t=2; t<n; t++) //next层数
for(int i=1; i<=size; i++) //prev
for(int j=1; j<=size; j++) //cur
for(int k=1; k<=size; k++){ //next
if((bit[i]&bit[k]) || (bit[j]&bit[k]) || (bit[k]&g[t])) continue;
f[t][j][k] = max(f[t][j][k], f[t-1][i][j]+cnt[k]);
}
//4.收集答案
int ans = 0;
for(int i=1; i<=size; i++)
for(int j=1; j<=size; j++)
ans = max(ans, f[n-1][i][j]);
cout<<ans<<'\n';
}
下面是某位大佬的ac代码(感觉思路一样,为啥我就wa了呢)
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
int n,m,s[102],cnt[102],tot;//s是可用集合,cnt是1的数目
int f[102][72][72];//f的2、3维都是可用状态的编号
//经试验m=10时|s|=60,所以可以放心少开
int dixing[102];//0平原,1山地,状态&地形!=0即炮兵在山地上
char str[13];
void input()//输入O(nm)
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i)
{
scanf("%s",str);
for(int j=0;j<m;++j)
{
dixing[i]<<=1;
if (str[j]=='H')
dixing[i]|=1;
}
}
}
int count1(int x)//计算1的个数
{
int sum=0;
while(x) x-=x&-x,++sum;
return sum;
}
void prework()//预处理合法集合O(2^m)
{
for(int i=0;i<1<<m;++i)
if (!(i&(i<<1))&&!(i&(i<<2)))
{
s[++tot]=i;
cnt[tot]=count1(i);//计算1的个数
}
}
int main()
{
input();
prework();
// for(int i=1;i<=n;++i)//DP O(n|s|^3)
// for(int j=1;j<=tot;++j)
// for(int k=1;k<=tot;++k)
// for(int l=1;l<=tot;++l)
// if ((s[j]&s[k])==0&&(s[j]&s[l])==0&&(s[j]&dixing[i])==0)
// f[i][j][k]=max(f[i-1][k][l]+cnt[j],f[i][j][k]);
// int ans=0;//寻找答案
// for(int i=1;i<=tot;++i)
// for(int j=1;j<=tot;++j)
// ans=max(f[n][i][j],ans);
cout<<tot<<endl;
// printf("%d",ans);
return 0;
}