LeetCode 361. 轰炸敌人（前缀和DP）

文章目录

- 1. 题目
- 2. 解题

1. 题目

想象一下炸弹人游戏，在你面前有一个二维的网格来表示地图，网格中的格子分别被以下三种符号占据：

'W' 表示一堵墙
'E' 表示一个敌人
'0'（数字 0）表示一个空位

在这里插入图片描述

请你计算一个炸弹最多能炸多少敌人。

由于炸弹的威力不足以穿透墙体，炸弹只能炸到同一行和同一列没被墙体挡住的敌人。

注意：你只能把炸弹放在一个空的格子里

示例:

输入: [["0","E","0","0"],["E","0","W","E"],["0","E","0","0"]]
输出: 3 
解释: 对于如下网格

0 E 0 0 
E 0 W E 
0 E 0 0

假如在位置 (1,1) 放置炸弹的话，可以炸到 3 个敌人

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/bomb-enemy
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

分别计算4个方向的前缀和，每个位置的数量是四个前缀和相加

class Solution {
public:
    int maxKilledEnemies(vector<vector<char>>& grid) {
    	if(grid.empty() || grid[0].empty())
    		return 0;
    	int m = grid.size(), n = grid[0].size(), i, j, pre = 0;
    	vector<vector<int>> ans(m, vector<int>(n,0));
    	//分四个方向求前缀和
    	for(i = 0; i < m; ++i)
    		for(j = 0, pre=0; j < n; ++j)
    		{
    			if(grid[i][j] =='W')
    				pre = 0;
    			else if(grid[i][j] =='E')
    				pre++;
    			else
    				ans[i][j] += pre;
    		}
    	for(i = 0; i < m; ++i)
    		for(j = n-1, pre=0; j >= 0; --j)
    		{
    			if(grid[i][j] =='W')
    				pre = 0;
    			else if(grid[i][j] =='E')
    				pre++;
    			else
    				ans[i][j] += pre;
    		}
    	for(j = 0; j < n; ++j)
    		for(i = 0, pre=0; i < m; ++i)
    		{
    			if(grid[i][j] =='W')
    				pre = 0;
    			else if(grid[i][j] =='E')
    				pre++;
    			else
    				ans[i][j] += pre;
    		}
    	for(j = 0; j < n; ++j)
    		for(i = m-1, pre=0; i >= 0; --i)
    		{
    			if(grid[i][j] =='W')
    				pre = 0;
    			else if(grid[i][j] =='E')
    				pre++;
    			else
    				ans[i][j] += pre;
    		}
    	int maxcount = 0;
    	for(i = 0; i < m; ++i)
    		for(j = 0; j < n; ++j)
    			maxcount = max(maxcount, ans[i][j]);
		return maxcount;
    }
};