YBTOJ炮兵阵地&&洛谷P2704
题目传送门
状压DP,详见代码
留给自己以后遇到题不会的模板
#include<bits/stdc++.h>
using namespace std;
int n,m,k;
int s[1005],g[1005];
int f[102][1005][1005],ans;
char ma[103];
int mp[103];
int get(int x)
{
int e=0;
while(x>0)
{
++e;
x-=x&(-x);
}
return e;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;++i)
{
scanf("%s",ma);
for(int j=0;j<m;++j)
if(ma[j]=='H') mp[i]+=1<<j;
}
for(int i=0;i<=(1<<m)-1;++i)
if(((i&(i<<1))==0)&&((i&(i<<2))==0)&&((i&(i>>1))==0)&&((i&(i>>2))==0))
{
++k;
s[k]=i;
g[k]=get(i);
if((i&mp[1])==0) f[1][0][k]=g[k];
}
for(int i=1;i<=k;++i)
for(int j=1;j<=k;++j)
if(((s[i]&s[j])==0)&&((s[j]&mp[2])==0)) f[2][i][j]=max(f[2][i][j],f[1][0][i]+g[j]);
for(int i=3;i<=n;++i)
for(int j=1;j<=k;++j)
if((mp[i]&s[j])==0)
for(int p=1;p<=k;++p)
if((s[p]&s[j])==0)
for(int q=1;q<=k;++q)
if(((s[q]&s[p])==0)&&((s[q]&s[j])==0)) f[i][p][j]=max(f[i][p][j],f[i-1][q][p]+g[j]);
for(int i=1;i<=k;++i)
for(int j=1;j<=k;++j)
ans=max(f[n][i][j],ans);
cout<<ans;
return 0;
}
在你像到洛谷上水双倍经验却MLE时,请往下看:
在洛谷上MLE之后,我把所有符合条件的状态都打出来了,发现只有60多种!然后把数组改小就过了……其实是不会写滚动数组
洛谷AC代码:
#include<bits/stdc++.h>
using namespace std;
int n,m,k,s[1005],g[1005],mp[103],f[102][70][70],ans;
char ma[103];
int get(int x)//计算某一状态含有多少个1(即有多少个炮兵)
{
int e=0;
while(x>0)
{
++e;
x-=x&(-x);
}
return e;
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;++i)//读入地图,将山地(不能放兵)的地方设为1
{
scanf("%s",ma);
for(int j=0;j<m;++j)
if(ma[j]=='H') mp[i]+=1<<j;
}
for(int i=0;i<=(1<<m)-1;++i)//枚举所有的状态
if(((i&(i<<1))==0)&&((i&(i<<2))==0)&&((i&(i>>1))==0)&&((i&(i>>2))==0))//判断每个1左右各两个是否有1,即判断这种状态是否存在
{
++k;
s[k]=i;
g[k]=get(i);
if((i&mp[1])==0) f[1][0][k]=g[k];//初始化第一行
}
//初始化第二行
for(int i=1;i<=k;++i)//枚举第一行状态
for(int j=1;j<=k;++j)//枚举第二行状态
if(((s[i]&s[j])==0)&&((s[j]&mp[2])==0)) f[2][i][j]=max(f[2][i][j],f[1][0][i]+g[j]);//判断是否与地形和第一行冲突
//dp过程
for(int i=3;i<=n;++i)//枚举当前行数
for(int j=1;j<=k;++j)//枚举当前行数状态
if((mp[i]&s[j])==0)//不与地形冲突
for(int p=1;p<=k;++p)//枚举前一行状态
if((s[p]&s[j])==0)//当前行状态不与前一行冲突
for(int q=1;q<=k;++q)//枚举前两行
//不与前两行冲突,且前两行自身不冲突
if(((s[q]&s[p])==0)&&((s[q]&s[j])==0)) f[i][p][j]=max(f[i][p][j],f[i-1][q][p]+g[j]);
for(int i=1;i<=k;++i)//枚举最后两行为结尾的情况,统计答案
for(int j=1;j<=k;++j)
ans=max(f[n][i][j],ans);
cout<<ans; //输出
return 0;
}