1. 程式人生 > >leecode [保持城市天際線]程式碼實現

leecode [保持城市天際線]程式碼實現

題目描述  https://leetcode-cn.com/problems/max-increase-to-keep-city-skyline/description/

  在二維陣列grid中,grid[i][j]代表位於某處的建築物的高度。 我們被允許增加任何數量(不同建築物的數量可能不同)的建築物的高度。 高度 0 也被認為是建築物。

最後,從新陣列的所有四個方向(即頂部,底部,左側和右側)觀看的“天際線”必須與原始陣列的天際線相同。 城市的天際線是從遠處觀看時,由所有建築物形成的矩形的外部輪廓

Python程式碼實現如下,如有錯誤遺漏,歡迎指正!

import pandas as pd
import numpy as np


def maxIncreaseKeepingSkyline( grid):
"""
:type grid: List[List[int]]
:rtype: int
"""
nums = grid
nums = np.array(nums).T
top_bottom = []
for i in range(nums.shape[0]):
ls = []
for j in range(nums.shape[1]):
ls.append(nums[i][j])
top_bottom.append(max(ls))

left_right = []
for i in range(nums.shape[1]):
ms = []
for j in range(nums.shape[0]):
ms.append(nums[j][i])
left_right.append(max(ms))
a = [[0 for _ in range(len(left_right))] for _ in range(len(top_bottom))]
a = np.array(a).T
cnt = 0
for i in range(len(left_right)):
for j in range(len(top_bottom)):
aa = min(left_right[i], top_bottom[j])
cnt = cnt + aa - nums[i][j]
a[i][j] = aa
return cnt


if __name__ == '__main__':
grid = [[3, 0, 8, 4], [2, 4, 5, 7], [9, 2, 6, 3], [0, 3, 1, 0]]
res = maxIncreaseKeepingSkyline(grid=grid)
print(res)