题目:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
提示:
n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105
分析:
参考力扣的解法,我把题目理解成,先建立三个列表:height 用于记录每个位置的柱高;leftheight用于记录每个位置左边的最高柱;rightheight用于记录每个位置右边的最高柱。由常识可以知道,储水量由两边的最短柱减去自身的高度就可以得到了。所以,本题只需要求出两边的最高柱即可。
核心数学公式:min(leftheight[i],rightheight[i])-height[i]
代码:
import numpy as np
import pandas as pd
import math
#coding='utf-8'
a = [0,1,0,2,1,0,1,3,2,1,2,1]
class Solution(object):
def __init__(self,height):
self.height = height
def trap(self):
leftMax = [0]
rightMax = []
sum = 0
n = len(self.height)
for i in range(1,n):
leftMax.append(max(leftMax[i-1],self.height[i]))
for i in range(1,n):
rightMax.append(0)
rightMax.append(1)
for i in range(n-2,-1,-1):
rightMax[i] = max(rightMax[i+1],self.height[i])
print(rightMax)
print("leftMax:",leftMax)
print("rightMax:",rightMax)
for i in range(n):
sum = sum+min(rightMax[i], leftMax[i]) - self.height[i]
print(sum)
solution1 = Solution(height=a) #每个位置对应的柱子高度
solution1.trap()