LeetCode 807. 保持城市天际线（Python）

在二维数组grid中，grid[i][j]代表位于某处的建筑物的高度。我们被允许增加任何数量（不同建筑物的数量可能不同）的建筑物的高度。高度 0 也被认为是建筑物。

最后，从新数组的所有四个方向（即顶部，底部，左侧和右侧）观看的“天际线”必须与原始数组的天际线相同。城市的天际线是从远处观看时，由所有建筑物形成的矩形的外部轮廓。请看下面的例子。

建筑物高度可以增加的最大总和是多少？

例子：
输入： grid = [[3,0,8,4],[2,4,5,7],[9,2,6,3],[0,3,1,0]]
输出： 35
解释： 
The grid is:
[ [3, 0, 8, 4], 
  [2, 4, 5, 7],
  [9, 2, 6, 3],
  [0, 3, 1, 0] ]

从数组竖直方向（即顶部，底部）看“天际线”是：[9, 4, 8, 7]
从水平水平方向（即左侧，右侧）看“天际线”是：[8, 7, 9, 3]

在不影响天际线的情况下对建筑物进行增高后，新数组如下：

gridNew = [ [8, 4, 8, 7],
            [7, 4, 7, 7],
            [9, 4, 8, 7],
            [3, 3, 3, 3] ]

说明:

1 < grid.length = grid[0].length <= 50。

grid[i][j] 的高度范围是： [0, 100]。

一座建筑物占据一个grid[i][j]：换言之，它们是 1 x 1 x grid[i][j] 的长方体。

思路：

暴力破解不解释。求出每一列、每一行的最大值，将二维数组填充，填充应满足不改变原来的每一行、每一列的最大值。

例如：每一列最大值为v = [9, 4, 8, 7]，每一行最大值为h = [8, 7, 9, 3]，则位于i，j处的数值应当取v[i]，和h[j]中较小的那一个

class Solution(object):
    def maxIncreaseKeepingSkyline(self, grid):
        """
        :type grid: List[List[int]]
        :rtype: int
        """
        vertical = list()
        horizontal = list()
        
        # 求每一列最大值
        for i in range(len(grid[0])):
            max_v = 0
            for j in range(len(grid)):
                if grid[j][i] > max_v:
                    max_v = grid[j][i]
                    
            vertical.append(max_v)
        
        # 求每一行最大值
        for i in range(len(grid)):
            max_h = 0
            for j in range(len(grid[i])):
                if grid[i][j] > max_h:
                    max_h = grid[i][j]
                    
            horizontal.append(max_h)
        
        # 计算需要增加的高度
        count = 0
        for i in range(len(grid)):
            for j in range(len(grid[i])):
                count = count + min(horizontal[i], vertical[j]) - grid[i][j]

        return count

LeetCode 807. 保持城市天际线（Python）

猜你喜欢