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++;elseans[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++;elseans[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++;elseans[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++;elseans[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;}
};