接雨水

給定 n 個(gè)非負(fù)整數(shù)表示每個(gè)寬度為 1 的柱子的高度圖，計(jì)算按此排列的柱子，下雨之后能接多少雨水。

輸入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
輸出：6
解釋：上面是由數(shù)組 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度圖，在這種情況下，可以接 6 個(gè)單位的雨水（藍(lán)色部分表示雨水）。
示例 2：

輸入：height = [4,2,0,3,2,5]
輸出：9

題解

python code

class Solution:def trap(self, height: List[int]) -> int:# 記錄柱子的個(gè)數(shù)length = len(height)# 記錄每個(gè)柱子左側(cè)最大高度l_list = list()# 記錄每根柱子右側(cè)最大高度r_list = list()# 默認(rèn)第一個(gè)柱子的 左側(cè)最大高度為0l_height = 0# 默認(rèn)最后一個(gè)柱子的 右側(cè)最大高度為0r_height = 0# 計(jì)算每根柱子左側(cè)的最大高度for val in height:if not l_list:l_list.append(l_height)l_height = valelse:l_list.append(l_height)if val > l_height:l_height = val# 計(jì)算每根柱子右側(cè)的最大高度for i in range(length-1, -1, -1):if not r_list:r_list.append(r_height)r_height = height[i]else:r_list.append(r_height)if height[i] > r_height:r_height = height[i]r_list = r_list[::-1]result = 0# 循環(huán)遍歷for k, v in enumerate(height):# 取左邊最大高度 和 右邊最大高度的最小值 減去 當(dāng)前元素的值# 如果小于0 則賦值為0 如果大于0 則為當(dāng)前柱子的存水量num = min(r_list[k], l_list[k]) - vval = num if num > 0 else 0result += valreturn result

總結(jié)

以上是生活随笔為你收集整理的python 接雨水的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。