这题单调栈存数组下标
遇arr[i] >=栈顶元素 arr[stack[len-1]] 则出栈stack[len-1]并和新栈顶计算存水量.持续循环直到栈为空或栈顶元素>arr[i],则入栈i
func trap(height []int) int {
if len(height) == 0 {
return 0
}
var res int
var stack []int
for i, h := range height {
l := len(stack)
// 这里2,1,0,1, 1>0,计算一次积水,1>=1计算一次积水(当然这次min(2,1)-1=0故计算结果为0,只是为了保证栈内没有两个连续重复数字罢了)
for l > 0 && h >= height[stack[l-1]] {
tIdx := stack[l-1]
stack = stack[:l-1]
l--
if l > 0 {
res += (min(height[stack[l-1]], h) - height[tIdx]) * (i - stack[l-1] - 1)
}
}
// 入栈的其实是下标而非元素值,目的是通过下标可以计算宽度
stack = append(stack, i)
}
return res
}
func min(i,j int) int {
if i < j {
return i
}
return j
}