leetcode（单调栈）- 接雨水 - 难度困难

1. 题目描述

给定 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
0 <= n <= 3 * 104
0 <= height[i] <= 105

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 题解

1、借助单调栈

单调栈：栈内元素是单调的（元素可以相同），本题用到的是递减栈。
用递减栈存入元素的下标，若当前元素大于栈顶，则出栈顶、计算雨水量，直至当前元素不大于栈顶时，将当前元素入栈。

/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function(height) {
    const stack = [];
    let water = 0;
    //特殊情况
    if(height.length < 3){
        return 0;
    }
    for(let i = 0; i < height.length; i++){
        while(stack.length && height[i] > height[stack[stack.length-1]]){
            //栈顶元素
            let popnum = stack.pop();
            //和栈顶相同的元素也出栈
            while(stack.length && height[stack[stack.length-1]] == height[popnum]){
                stack.pop();
            }
            //计算该层的雨水
            if(stack.length){
                let temp = height[stack[stack.length-1]];
                //高度
                let high = Math.min(temp-height[popnum], height[i]-height[popnum]);
                //宽度
                let width = i - stack[stack.length-1]-1;
                water += high * width;
            }
        }
        //入栈的是索引
        stack.push(i);
    }
    return water;
};

参考：https://leetcode-cn.com/problems/trapping-rain-water/solution/yi-miao-jiu-neng-du-dong-de-dong-hua-jie-o9sv/