go语言LeetCode刷题笔记
Hot100系列
第42题接雨水
-
给定
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 = [0,1,0,2,1,0,1,3,2,1,2,1]
那么假设我们要计算索引为6的格子能存储多少水,该怎么算呢?
我们观察图中,可以知道。存储水的量依赖于该格子的高度、该格子左边的最大高度,以及该格子右边的最大高度。
当左右的最大高度的最小值大于该格子的最大高度时,可以储水。
当左右的最大高度的最小值小于该格子的最大高度时,不可以储水。(可以看上图的index=1、3等柱子为例子)
那么单个格子(柱子hhh)的计算公式应该是什么样子的呢?
C
=
m
i
n
(
l
e
f
t
[
i
]
,
r
i
g
h
t
[
i
]
)
∗
h
e
i
g
h
t
[
i
]
C = min(left[i], right[i]) * height[i]
C=min(left[i],right[i])∗height[i]
当然上述公式要满足左右的最大高度的最小值大于该格子的最大高度才能成立。
go语言实现代码如下
func trap(height []int) int {
//每一个位置的单位储水量由该单位的右边最大值和该单位的左边最大值之间的最小值,以及该单位的高度决定
n := len(height)
left := make([]int, n) //记录左边最大数组
right := make([]int, n) //记录右边最大数组
left[0] = height[0] //初始化,避免数组越位
right[n-1] = height[n-1] //初始化,避免数组越位
for i := 1; i < n; i++ { //初始化左右最大值数组
if height[i] > left[i-1] {
left[i] = height[i]
}else{
left[i] = left[i-1]
}
if height[n-i-1] > right[n-i] {
right[n-i-1] = height[n-i-1]
}else{
right[n-i-1] = right[n-i]
}
}
sum := 0
for i := 0; i < n; i++ { //计算蓄水量总和
if min(left[i], right[i]) > height[i] {
sum += min(left[i], right[i]) - height[i]
}
}
return sum
}
func min(a, b int) int {if a < b {return a}; return b}