LeetCode 42.接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

---

单调栈

当前柱子比栈顶高时，判断是否可以接到雨水

class Solution{
public:
    int trap(vector<int> &height)
    {
        stack<int> s;
        int ans = 0;
        for (int i = 0; i < height.size(); ++i)
        {
            while (!s.empty() && height[i] > height[s.top()]) //比上一个柱子高，可能会接到雨水
            {
                int now = s.top(); //栈顶柱子的位置
                s.pop();
                if (s.empty())
                    break;         //当前就只有两个柱子，接不到雨水比如0，1
                int pre = s.top(); //前一个柱子的位置
                if (height[pre] == height[now])
                    continue;      //相邻两个柱子等高，接不到水
                int w = i - pre - 1, h = min(height[i], height[pre]); //能接到水的宽度和高度
                ans += (h - height[now]) * w;
            }
            s.push(i);
        }
        return ans;
    }
};

保存当前柱子的左右柱子最高点，计算时取左右的最小值，如果当前柱子减去最小值大于0说明能接到雨水

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.empty())
            return 0;
        int len = height.size();
        vector<int> right(len, 0), left(len, 0);
        
        for (int i = 1; i < len; ++i)
            left[i] = max(left[i - 1], height[i - 1]);//计算左边最高点
        for (int i = len - 2; ~i; --i)
            right[i] = max(right[i + 1], height[i + 1]);//右边
        
        int ans = 0;
        for (int i = 0; i < len; ++i)
        {
            int h = min(left[i], right[i]);
            ans += max(0, h - height[i]);//大于0，能接到雨水
        }
        return ans;
    }
};