LeetCode-Java：42.接雨水

题目

给定 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

思路

根据图示可知，数组的第一个和最后一个肯定无法接水，计算的时候可以去掉。

数组正序遍历，只考虑当前柱子是否比上一个高，在示例1中会呈现为：

即多出浅蓝色三个格子。

class Solution {
    public int trap(int[] height) {
        // 错误代码：会导致形成上图多3个格子的情况
        int water=0;
        int index=0;
        boolean jump=true;
        for(int i=1;i<height.length-1;i++){
            //第一个和最后一个肯定无法接水，跳过
            if(height[index]>height[i]){
                if(jump==true){
                    //如果jump是真，表示还没有遇到一个柱子比上一个柱子高
                    index=i-1;
                    jump=false;
                }
                water+=height[index]-height[i];
            }
            else{
                jump=true;
                index=i;
            }
             System.out.print("/"+water);
        }
        return water;
    }
}

所以，应当先找到height[i-1]>height[i]，记录index=i-1，再从i开始向后找到1个height[i]>=height[index]，找到后统计这部分内的蓄水量。

class Solution {
    public int trap(int[] height) {
        // 错误代码，对数组正序、倒序遍历，找到第一个i-1高于i的，再找下一个大于i-1的
        // 没考虑到无法从i开始继续找到一个比i-1高的
        int water=0;
        int index=0;
        boolean jump=true;
        for(int i=1;i<height.length-1;i++){
            //第一个和最后一个肯定无法接水，跳过
            if(height[index]>height[i]){
                if(jump==true){
                    index=i-1;
                    jump=false;
                }
            }
            else{
                // 找到index和i二者之间较小的作为高
                int min=(height[index]>height[i])?height[i]:height[index];
                for(int j=index+1;j<i;j++){
                    water+=min-height[j];
                }
                index =i;
                jump=true;
            }
            if(height[i-1]>height[i])
            System.out.print("/"+water);
        }
        index=height.length-1;
        jump=true;
        for(int i=height.length-2;i>0;i--){
            if(height[index]>height[i]){
                if(jump==true){
                    index=i+1;
                    jump=false;
                }
           }
           else{
            int min=(height[index]>height[i])?height[i]:height[index];
            for(int j=index-1;j>i;j--){
                water+=min-height[j];
            }
            index=i;
            jump=true;
           }
        }
        return water;
    }
}

但，还是会存在找不到比i-1高的情况，比如[4,2,0,3,2,5]，无论是从右开始还是从左开始都无法找到，最终输出的蓄水量是0。

这道题没想到合适的思路，查看了题解之后了解到 单调栈，尝试还原单调栈的思路。

单调栈相关知识

java.util.Stack 常用方法

1. push(E item): 将指定的元素推入此栈的顶部。

   Stack<String> stack = new Stack<>();
   stack.push("A");
   stack.push("B");
   

2. pop(): 移除此栈顶部的元素，并将其返回。

   String poppedElement = stack.pop();
   

3. peek(): 查看此栈顶部的元素，但不移除它。

   String topElement = stack.peek();
   

4. empty(): 判断栈是否为空。

   if (!stack.empty()) {
       // 栈不为空
   }
   

5. search(Object o): 查找指定元素在栈中的位置（从栈顶开始计数，1 表示栈顶元素）。

   int position = stack.search("A");
   

单调栈

单调栈是一种特殊的栈数据结构，栈中的元素按照单调递增或单调递减的顺序排列。

单调递增栈：栈顶元素最小，栈底最大；当当前元素大于栈顶时，栈顶元素出栈，并进行相关操作，栈必须保持单调递增的状态。

单调递减栈：栈顶元素最大，栈底最小。

解

① 单调栈

运行时间13ms，超过19.66%；内存消耗44.58MB，超过83.14%

这里的难点在于出栈时如何计算雨水面积

当出现栈顶元素小于当前元素的高度时，根据单调栈的要求，如果入栈的话不满足单调递增（从栈顶到栈底），此时需要出栈。

出栈时，以当前i作为右侧墙壁，栈顶的下一个作为左侧墙壁，左右墙壁较小的那个减去当前元素的高度，可得出高度high，此时还需要乘以底（即左右墙壁之间隔了多少个格子，right-left-1，这里的right=i）

import java.util.Stack;
class Solution {
    public int trap(int[] height) {
        // 单调栈
        Stack<Integer> stack = new Stack<>();
        int water=0;
        for(int i=0;i<height.length;i++){
            while(!stack.empty() && height[stack.peek()]<height[i]){
                //如果栈非空，并且栈顶元素大于i，则出栈
                int index = stack.pop();//移除栈顶元素并返回对应的下标
                if(stack.empty()) break;
                int left=stack.peek();//获得当前的栈顶元素
                //当前栈顶元素是左侧，i是右侧，可以填充的高度应该为 左右侧较低的那个 减去当前高度
                int high=(height[left]>height[i])?height[i]:height[left];
                high-=height[index];
                water+=high*(i-left-1);
            }
            stack.push(i);         
        }
        return water;
    }
}