LeetCode42. 接雨水(day0010_2)

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

• 1 <= n <= 2 * 104

• 0 <= height[i] <= 105

双指针法按列：

class Solution {
public:
    int trap(vector<int>& height) {
        int sum=0;
        for(int i=0;i<height.size();i++){
            //最前和最后不接雨水
            if(i==0 || i==height.size()-1) continue;
            //定义双指针
            int rheight=height[i];//记录右侧最高
            int lheight=height[i];//记录左侧最高
            //遍历找到i列右侧最高的柱子
            for(int r=i+1;r<height.size();r++){
                if(height[r]>rheight) rheight=height[r];
            }
            //遍历找到i列左侧最高的柱子
            for(int l=i-1;l>=0;l--){
                if(height[l]>lheight) lheight=height[l];
            }
            //取左右侧最高柱子中，相对较小的，测量接水最佳高度
            int h=min(lheight,rheight)-height[i];
            if(h>0) sum+=h;
        }
        return sum;
    }
};

动态规划解法:

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size()<=2) return 0;
        //定义结果变量
        int sum=0;
        //定义两个左右存最大最小值的容器
        vector<int>maxLeft(height.size(),0);
        vector<int>maxRight(height.size(),0);
        //定义一下大小,简洁后面代码
        int size=maxRight.size();

        //记录每个左边柱子的最大高度
        maxLeft[0]=height[0];
        for(int i=1;i<size;i++){
            maxLeft[i]=max(height[i],maxLeft[i-1]);
        }
        //记录每个右边柱子的最大高度
        maxRight[size-1]=height[size-1];
        for(int i=size-2;i>=0;i--){
            maxRight[i]=max(height[i],maxRight[i+1]);
        }

        //求和
        for(int i=0;i<size;i++){
            int count=min(maxLeft[i],maxRight[i])-height[i];
            if(count>0)sum+=count;
        }
        return sum;
    }
};

单调栈解法：

class Solution {
public:
    int trap(vector<int>& height) {
        //小于等于两个没办法接水
        if(height.size()<=2)return 0;
        //存放下标，计算的时候使用下标对应的柱子高度
        stack<int>st;
        //存放第一个柱子下标
        st.push(0);
        //定义和
        int sum=0;
        //遍历柱子，解决面对的三种情况
        for(int i=1;i<height.size();i++){
            //情况一（高度小于栈顶高度）
            if(height[i]<height[st.top()]){
                st.push(i);
            }
            //情况二（高度等于栈顶高度）
            if(height[i]==height[st.top()]){
                st.pop();
                st.push(i);
            }
            //情况三（高度大于栈顶高度）
            else{
                while(!st.empty()&&height[i]>height[st.top()]){
                    //定义最低槽
                    int mid=st.top();
                    st.pop();
                    if(!st.empty()){
                        //找到左右最大高中较小的，计算存水高度
                        int h=min(height[st.top()],height[i])-height[mid];
                        //计算宽度
                        int w=i-st.top()-1;
                        sum+=h*w;
                    }
                }
                st.push(i);
            }
        }
        return sum;
    }
};