LeetCode 361. 轰炸敌人(前缀和DP)

文章目录

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;
    }
};

36 ms 9.6 MB


我的CSDN博客地址 https://michael.blog.csdn.net/

长按或扫码关注我的公众号(Michael阿明),一起加油、一起学习进步!
Michael阿明

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/107527864