Leetcode：42. 接雨水（单调栈C++）

目录

42. 接雨水

题目描述：

实现代码与解析：

单调栈

原理思路：

---

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

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

实现代码与解析：

单调栈

class Solution {
public:
    int trap(vector<int>& height) {
        int res = 0;
        stack<int> st;

        for (int i = 0; i < height.size(); i++)
        {
            while (st.size() && height[i] > height[st.top()])
            {
                int mid = st.top(); // 中间
                st.pop();
                if (st.empty()) break; // 需要有左侧墙
                int l = st.top(); // 左侧
                int w = i - l - 1; // 宽度
                int h = min(height[l], height[i]) - height[mid]; // 高度为左右墙中最低的减去中间的
                res += w * h;
            }
            st.push(i);
            //循环过后，左侧的墙就会变为下一次中间的墙，从而填满其他层
        }
    return res;
    }
};

原理思路：

        单调栈，和一般的单调栈差不多，但是这里栈中必须要有两个元素来构成左侧和在中间，如果只有一个元素，那么不能装雨水，右侧墙就为第一个比其大的值，

        和下面单调栈的题同。Leetcode：496. 下一个更大元素 I、503. 下一个更大元素 II（单调栈C++）_Cosmoshhhyyy的博客-CSDN博客

        宽度就为其左侧和右侧的差值+1，高度显然就为左右两侧的最小值减去中间，然后相乘算出此次两侧墙高度差能装的体积。

        大家最好根据代码和例子的图片带入演绎一遍，可以更好的理解，光看代码一上来确实有点懵。