leetcode42. 接雨水-小羊的记录本

小羊羊羊羊mie

于 2020-07-28 01:05:24 发布

阅读量132

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode 栈

本文链接：https://blog.csdn.net/weixin_40593830/article/details/107625740

leetcode 专栏收录该内容

53 篇文章

订阅专栏

本文深入解析了经典的接雨水问题，采用单调栈算法高效求解。通过实例[0,1,0,2,1,0,1,3,2,1,2,1]，详细阐述了如何计算柱状图中可容纳的雨水量，输出结果为6单位。文章重点介绍了单调栈的使用技巧，包括处理边界条件的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

42.接雨水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
在这里插入图片描述
上面是由数组 [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

思路：
1.单调栈：等到下一个数大于前一个数（栈顶的数），弹出当前的栈顶（凹槽），通过当前节点，当前栈顶元素，弹出的元素这三者来分别计算长和宽，计算面积；
2.注意栈是空的情况：左边或者右边没有数，也盛不了水；
考虑[1,0,0,1]的入栈过程，有助于理解while语句；

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