想象一下炸弹人游戏,在你面前有一个二维的网格来表示地图,网格中的格子分别被以下三种符号占据:
'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 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
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;
}
};
36 ms 9.6 MB