单调栈:Leetcode面试题17.21直方图的水量

这题单调栈存数组下标

遇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
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值