LeetCode 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
0 <= n <= 3 * 104
0 <= height[i] <= 105

对于给定的height[i]，找出它左边最高的柱子和右边最高的柱子就行。
我是用单调栈写的，有点麻烦了。开两个数组分别遍历一遍就行了。

class Solution {
public:
    static const int maxn = 3e4 + 10;
    int lef[maxn];
    int rig[maxn];
    int st[maxn];//单调栈
    int top = 0;
    int trap(vector<int>& height) {
        int ans = 0, tem = 0;
        for(int i = 0; i < height.size(); i++) {//找出每个柱子左边最高的柱子，省事点可以直接遍历
            while(top && height[i] > st[top])   top--;
            lef[i] = st[top];
            if(height[i] > st[top])
                st[++top] = height[i];
        }
        top = 0;
        for(int i = height.size() - 1; i >= 0; i--) {//找出每个柱子右边最高的柱子
            while(top && height[i] > st[top])   top--;
            rig[i] = st[top];
            if(height[i] > st[top])
              st[++top] = height[i];
        }
        for(int i = 0; i < height.size(); i++) {
            ans += max(0, min(lef[i], rig[i]) - height[i]);
        }
        return ans;
    }
};