mirror of
https://github.com/chefyuan/algorithm-base.git
synced 2024-11-16 17:13:39 +00:00
118 lines
5.8 KiB
Markdown
118 lines
5.8 KiB
Markdown
|
### leetcode 42 接雨水
|
|||
|
|
|
|||
|
|
这道接雨水也是一道特别经典的题目，一道必刷题目，我们也用单调栈来解决。下面我们来看一下题目吧
|
|||
|
|
|
|||
|
|
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
|
|||
|
|
|
|||
|
|
示例 1：
|
|||
|
|
|
|||
|
|
```
|
|||
|
|
输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
|
|||
|
|
输出：6
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
示例 2：
|
|||
|
|
|
|||
|
|
```
|
|||
|
|
输入：height = [4,2,0,3,2,5]
|
|||
|
|
输出：9
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
示例3：
|
|||
|
|
|
|||
|
|
```
|
|||
|
|
输入：[4,3,2,0,1,1,5]
|
|||
|
|
输出：13
|
|||
|
|
```
|
|||
|
|
|
|||
|
|
> 上面是由数组 [4,3,2,0,1,1,5]表示的高度图，在这种情况下，可以接 13个单位的雨水（见下图）。
|
|||
|
|
|
|||
|
|
### 题目解析：
|
|||
|
|
|
|||
|
|
看了上面的示例刚开始刷题的同学可能有些懵逼，那我们结合图片来理解一下，我们就用示例3的例子进行举例，他的雨水到底代表的是什么。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
上图则为我们的题目描述，是不是理解了呢？你也可以这样理解我们在地上放置了若干高度的黄色箱子，他们中间有空隙，然后我们想在他们里面插入若干蓝色箱子，并保证插入之后，这些箱子的左视图和右视图都不能看到蓝色箱子。
|
|||
|
|
|
|||
|
|
好啦题目我们已经理解了，下面我们看一下解题思路。做这个这前我们可以先去看一下我们之前做过的另一道题目每日温度。这两道题目的思路差不多，都是利用了单调栈的思想，下面我们来看一下具体思路吧。
|
|||
|
|
|
|||
|
|
这里我们也系统的说一下单调栈，单调栈含义就是栈内的元素是单调的，我们这两个题目用到的都是递减栈（相同也可以），我们依次将元素压入栈，如果当前元素小于等于栈顶元素则入栈，如果大于栈顶元素则先将栈顶不断出栈，直到当前元素小于或等于栈顶元素为止，然后再将当前元素入栈。就比如下图的4，想入栈的话则需要2，3出栈之后才能入栈，因为4大于他俩。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
我们了解单调栈的含义下面我们来看一下接雨水问题到底该怎么做，其实原理也很简单，我们通过我们的例3来进行说明。
|
|||
|
|
|
|||
|
|
首先我们依次入栈4，3，2，0我们的数组前四个元素是符合单调栈规则的。但是我们的第五个1，是大于0的。那我们就需要0出栈1入栈。但是我们这样做是为了什么呢？有什么意义呢？别急我们来看下图。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
上图我们的，4，3，2，0已经入栈了，我们的另一个元素为1，栈顶元素为0，栈顶下的元素为2。那么我们在这一层接到的雨水数量怎么算呢？2，0，1这三个元素可以接住的水为一个单位(见下图)这是我们第一层接到水的数量。
|
|||
|
|
|
|||
|
|
注：能接到水的情况，肯定是中间低两边高，这样才可以。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
因为我们需要维护一个单调栈，所以我们则需要将0出栈1入栈，那么此时栈内元素为4，3，2，1。下一位元素为1，我们入栈，此时栈内元素为4，3，2，1，1。下一元素为5，栈顶元素为1，栈顶的下一元素仍为1，则需要再下一个元素，为2，那我们求当前层接到的水的数量。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
我们是通过2，1，1，5这四个元素求得第二层的接水数为1*3=3;1是因为min(2-1,5-1)=min(1,4)得来的，大家可以思考一下木桶效应。装水的多少，肯定是按最短的那个木板来的，所以高度为1，3的话是因为5的索引为6，2的索引为2，他们之间共有三个元素（3，4，5）也就是3个单位。所以为6-2-1=3。
|
|||
|
|
|
|||
|
|
将1出栈之后，我们栈顶元素就变成了2，下一元素变成了3，那么3，2，5这三个元素同样也可以接到水。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
这是第三层的接水情况，能够接到4个单位的水，下面我们继续出栈2，那么我们的4，3，5仍然可以接到水啊。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
这是我们第四层接水的情况，一共能够接到5个单位的水，那么我们总的接水数加起来，那就是
|
|||
|
|
|
|||
|
|
1+3+4+5=13。你学会了吗？别急还有视频我们，我们再来深入理解一哈。
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
### 题目代码：
|
|||
|
|
|
|||
|
|
```java
|
|||
|
|
class Solution {
|
|||
|
|
public int trap(int[] height) {
|
|||
|
|
Stack<Integer> stack = new Stack<Integer>();
|
|||
|
|
int water = 0;
|
|||
|
|
//特殊情况
|
|||
|
|
if(height.length <3){
|
|||
|
|
return 0;
|
|||
|
|
}
|
|||
|
|
for(int i = 0; i < height.length; i++){
|
|||
|
|
while(!stack.isEmpty() && height[i] > height[stack.peek()]){
|
|||
|
|
//栈顶元素
|
|||
|
|
int popnum = stack.pop();
|
|||
|
|
//相同元素的情况例1，1
|
|||
|
|
while(!stack.isEmpty()&&height[popnum] == height[stack.peek()]){
|
|||
|
|
stack.pop();
|
|||
|
|
}
|
|||
|
|
//计算该层的水的单位
|
|||
|
|
if(!stack.isEmpty()){
|
|||
|
|
int temp = height[stack.peek()];//栈顶元素值
|
|||
|
|
//高
|
|||
|
|
int hig = Math.min(temp-height[popnum],height[i]-height[popnum]);
|
|||
|
|
//宽
|
|||
|
|
int wid = i-stack.peek()-1;
|
|||
|
|
water +=hig * wid;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
//这里入栈的是索引
|
|||
|
|
stack.push(i);
|
|||
|
|
}
|
|||
|
|
return water;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
```
|
|||
|
|
|