p2074 炮兵阵地-洛谷-题解-状态压缩

传送门：

P2704 [NOI2001] 炮兵阵地 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)https://www.luogu.com.cn/problem/P2704

---

题目描述:

司令部的将军们打算在NM的网格地图上部署他们的炮兵部队。一个NM的地图由N行M列组成，地图的每一格可能是山地（用“H” 表示），也可能是平原（用“P”表示），如下图。在每一格平原地形上最多可以布置一支炮兵部队（山地上不能够部署炮兵部队）；一支炮兵部队在地图上的攻击范围如图中黑色区域所示：

如果在地图中的灰色所标识的平原上部署一支炮兵部队，则图中的黑色的网格表示它能够攻击到的区域：沿横向左右各两格，沿纵向上下各两格。图上其它白色网格均攻击不到。从图上可见炮兵的攻击范围不受地形的影响。
　　现在，将军们规划如何部署炮兵部队，在防止误伤的前提下（保证任何两支炮兵部队之间不能互相攻击，即任何一支炮兵部队都不在其他支炮兵部队的攻击范围内），在整个地图区域内最多能够摆放多少我军的炮兵部队。

Input

第一行包含两个由空格分割开的正整数，分别表示N和M；
　　接下来的N行，每一行含有连续的M个字符（‘P’或者‘H’），中间没有空格。按顺序表示地图中每一行的数据。N≤100；M≤10。

Output

仅在第一行包含一个整数K，表示最多能摆放的炮兵部队的数量。

Sample Input

5 4
PHPP
PPHH
PPPP
PHPP
PHHP

---

        这道题真的毒瘤啊，一开始用java去做，搞了半天一直超时，本蒟蒻以为是因为java本来就慢，然后就用c++写了一边。结果。。。c++也超时，然后仔细优化，后来发现多写几个跳出循环的条件就可以减少大量代码运行时间。

       原代码：(这里先不给解析，主要看看我那几个continue）

for(int i=2;i<=n;i++){
            for(int s=0;s<len;s++){
                int t=b[s];
                if((s<<2&s)>0||(s<<1&s)>0) continue;//第一个continue
                if((a[i]|s)!=a[i]) continue;        //第二个continue
                for(int p=0;p<len;p++){
                    if((s&p)!=0) continue;        //第三个个continue
                    for(int q=0;q<len;q++){
                        if((s&q)!=0) continue;    //第四个continue
                        dp[i%3][s][p]=Math.max(dp[i%3][s][p],dp[(i-1)%3][p][q]+t);
                        if(i==n) ans=Math.max(ans,dp[i%3][s][p]);
                    }
                }
            }
        }

        改进后代码:(大家可以发现判断条件多了一些)

for(int i=2;i<=n;i++){
		for(int s=0;s<len;s++){
			if((s<<1&s)||(s<<2&s)) continue;//第一个和第二个continue没变
			if((s|a[i])!=a[i]) continue;    //本蒟蒻没有想到那还可以变，太烧脑了
			for(int p=0;p<len;p++){
				if(s&p||((a[i-1]|p)!=a[i-1])||(p<<1&p)||(p<<2&p)) continue;
                            //这里加了很多判断条件，可以大幅减少循环次数
				for(int q=0;q<len;q++){
					if(s&q||(q<<1&q||q<<2&q)) continue;//这里也加了些判断条件
					dp[i%3][s][p]=max(dp[i%3][s][p],dp[(i-1)%3][p][q]+b[s]);
				}
			}
		}
	}

        不难想出来，多弄写判断条件可以大幅减少不必要的循环，当时没有想到这样优化，长记性了!!!

完整代码:(附解释，c++和java都能AC100）

#include<iostream>
using namespace std;
int n,m;
int a[101],b[1<<10];    //a数组记录每一行地形的状态，例如PHHP记录为1001，1代表平原
int dp[3][1<<10][1<<10];//b数组记录每一种情况的人数，录入1001记录为2(人数为2），1代表士兵
inline int f(int num){  //该函数用于求二进制数中含1的个数，例如100111含有4个1
	int sum=0;
	while(num){
		if(num&1) sum++;
		num>>=1;
	}
	return sum;
}
int main()
{
	cin>>n>>m;
	char ch;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			cin>>ch,a[i]<<=1,a[i]+=(ch=='P'?1:0); //小处理一下
	int len=1<<m;
	for(int i=0;i<len;i++)      //我们用b数组记录下来f函数的结果，方便后面使用，防止爆时间
		b[i]=f(i);
	for(int s=0;s<len;s++){
		if((s<<1&s)||(s<<2&s)) continue;//炮兵的横向攻击范围有两个距离单位，纵向这里不需要判断
		if((s|a[1])!=a[1]) continue;    //炮兵只能站在平原上
		dp[1][s][0]=b[s];                  
	}
	int ans=0;
	for(int i=2;i<=n;i++){
		for(int s=0;s<len;s++){
			if((s<<1&s)||(s<<2&s)) continue;
            //炮兵的横向攻击范围有两个距离单位，纵向在下面判断
			if((s|a[i])!=a[i]) continue; 
            //炮兵只能站在平原上
			for(int p=0;p<len;p++){
				if(s&p||((a[i-1]|p)!=a[i-1])||(p<<1&p)||(p<<2&p)) continue;
                //p为i-1行炮兵情况
                //s&p:i行和i-1行的炮兵不能打到对方
                //((a[i-1]|p)!=a[i-1]):i-1行的所有炮兵只能站在平原上
                //(p<<1&p)||(p<<2&p):i-1行的所有炮兵不能攻击到横向的自己人
				for(int q=0;q<len;q++){
					if(s&q||(q<<1&q||q<<2&q)) continue;
                    //同上
					dp[i%3][s][p]=max(dp[i%3][s][p],dp[(i-1)%3][p][q]+b[s]);
                    //我们这里用滚动数组，要不然很可能爆内存，这个不难，就不解释了
				}
			}
		}
	}
	for(int s=0;s<len;s++)
		for(int p=0;p<len;p++)
			ans=max(ans,dp[n%3][s][p]);
	cout<<ans<<endl;
	return 0;
}