LeetCode-Python-361. 轰炸敌人-CSDN博客

本文链接：https://blog.csdn.net/qq_32424059/article/details/90415006

给你一个二维的网格图，网格图中的每一个格子里要么是一堵墙 '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 个敌人。

思路：

找到所有是0的点，然后向四个方向找可以炸死的敌人

class Solution(object):
    def maxKilledEnemies(self, grid):
        """
        :type grid: List[List[str]]
        :rtype: int
        """
        if not grid or not grid[0]:
            return 0
        m, n = len(grid), len(grid[0])
        res = 0
        def count(x0, y0):
            cnt = 0
            for x in range(x0, -1, -1): #向上
                if grid[x][y0] == "E":
                    cnt += 1
                elif grid[x][y0] == "W":
                    break
                    
            for x in range(x0, m):  #向下
                if grid[x][y0] == "E":
                    cnt += 1
                elif grid[x][y0] == "W":
                    break
                    
            for y in range(y0, -1, -1): #向左
                if grid[x0][y] == "E":
                    cnt += 1
                elif grid[x0][y] == "W":
                    break      
                     
            for y in range(y0, n): #向右
                if grid[x0][y] == "E":
                    cnt += 1
                elif grid[x0][y] == "W":
                    break   
                    
            return cnt
            
        for i in range(m):
            for j in range(n):
                if grid[i][j] == "0":
                    res = max(res, count(i, j))
        return res