标签：peek int res 雨水 stk leetcode42 height Math

接雨水

• dp

维护两个dp数组，分别记录左侧和右侧的最大值。当前位置的雨水就是左右两侧最大值的较小值减去当前位置的高度，将所有雨水累加即可得到结果。

class Solution {
    public int trap(int[] height) {
        int n = height.length, left[] = new int[n], right[] = new int[n], res = 0;
        for(int i = 0; i < n; i++){
            left[i] = i == 0 ? height[0] : Math.max(left[i-1], height[i]);
            right[n-1-i] = i == 0 ? height[n-1] : Math.max(right[n-i], height[n-1-i]);
        }
        for(int i = 0; i < n; i++)
            res += Math.min(left[i], right[i])-height[i];
        return res;
    }
}

• 单调栈

1. dp方法相当于竖直方向切分，单调栈方法相当于横向切分。
2. 注意只有“高-矮-高”这种形式才可以接住雨水。
3. 需要维护一个递减栈，从0开始进行遍历。如果栈不为空且栈顶元素小于当前元素，则pop出一个元素。pop出元素后，如果栈还不为空，那么满足了“高-矮-高”这种特性，横向切分后的高度等于两边较小值减去中间的较低值，即Math.min(height[stk.peek()], height[i])-height[index]，宽度为两条高边的下标之差减一，即i-stk.peek()-1。如果pop出元素后栈为空，那么只能满足“矮-高”这种情况，无法存储雨水，则直接跳出循环。
4. 循环结束后，将当前的下标存入栈中。

class Solution {
    public int trap(int[] height) {
        Deque<Integer> stk = new ArrayDeque<>();
        int res = 0, n = height.length;
        for(int i = 0; i < n; i++){
            while(!stk.isEmpty() && height[stk.peek()] < height[i]){
                int index = stk.pop();
                if(stk.isEmpty())   break;
                int w = i-stk.peek()-1, h = Math.min(height[stk.peek()], height[i])-height[index];
                res += w*h;
            }
            stk.push(i);
        }
        return res;
    }
}

标签：peek,int,res,雨水,stk,leetcode42,height,Math
来源： https://www.cnblogs.com/xzh-yyds/p/16598356.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。