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

问题分析

此题用单调栈解决是最优解法。从前往后遍历数组，每次取得元素，判断如果当前栈为空或者当前数小于等于栈顶的数，那么将当前的数的下标压入栈；判断如果当前取得的元素大于栈顶元素，说明有可能有接水的坑出现，将栈顶元素记录为临时变量表示坑底，并弹出。然后判断如果这时栈为空，则说明没有左边界没有坑接水，那么continue准备将当前元素压入栈。如果这时栈不为空，那么利用当前栈顶元素、当前元素和记录坑底的临时变量，计算出接水的多少，并加到ans中。当遍历完整个数组后，ans就是整个接水的量，返回即可。

代码实现

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> sta;
        int ans = 0;
        int i = 0;
        while(i < height.size()){
            if(sta.empty() || height[i] <= height[sta.top()]){
                sta.push(i);
                i++;
            }
            else{
                int temp = sta.top();
                sta.pop();
                if(sta.empty())
                    continue;
                ans += (min(height[sta.top()], height[i]) - height[temp]) * (i - sta.top() - 1);
            }
        }
        return ans;
    }
};