给定
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 个单位的雨水(蓝色部分表示雨水)。
单调栈:栈内元素单调按照递增(递减)顺序排列的栈
func trap(height []int) (ans int) {
stack := []int{} //使用切片模拟单调栈
for i, h := range height {
//循环遍历单调栈中的下标,求每个下标对应的面积
for len(stack) > 0 && h > height[stack[len(stack)-1]] {
//栈顶元素下标
top := stack[len(stack)-1]
stack = stack[:len(stack)-1]
if len(stack) == 0 {
break
}
//栈顶的下一个元素下标
left := stack[len(stack)-1]
//求到这个下标的低洼处的宽度和高度
curWidth := i - left - 1
curHeight := min(height[left], h) - height[top]
ans += curWidth * curHeight
// fmt.Printf("%d -- %d -- %d -- %d \n", left, curWidth, curHeight, ans)
}
stack = append(stack, i)
}
return
}
func min(a, b int) int {
if a < b {
return a
}
return b
}
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。