187、【栈与队列】leetcode ——42. 接雨水（C++版本）

题目描述

原题链接：42. 接雨水

解题思路

（1）暴力法

每遍历到一个高度时，就左右寻找探索两侧的边界，然后取二者中的最小值与当前高度相减（取最小值的原因，相当于是一个最短的木桶板决定了它最大可存水量的上界）。

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size(), res = 0;
        for(int i = 0; i < n; i++) {
        	// 遇到第一个和最后一个高度不左右探索
            if(i == 0 || i == n - 1)            continue;
            int left = height[i], right = height[i];
            // 寻找左边界
            for(int l = i - 1; l >= 0; l--) {
                if(height[l] > left)        left = height[l];
            }
            // 寻找右边界
            for(int r = i + 1; r < n; r++) {
                if(height[r] > right)       right = height[r];
            }
            // 找到可存水量之差
            int h = min(left, right) - height[i];
            if(h > 0)       res += h;
        }  

        return res;
    }
};

此方式会超时。

（2）单调栈

单调栈主要用于解决找到某一侧遇到的第一个更大或更小的值，因此该题可用单调栈来存储来”未达到封口凹槽“的一边。

存储元素：数组下标，存储次序：从栈底到栈顶，单调不增（包含递减和等于的情况）。

由于目标是让求出接多少雨水，也就相当于是求可接雨水体积。用单调栈来存储高度值，当为单调不增时入栈，当遇到的数比栈顶元素大时，就把栈顶元素弹出，计算当前栈顶元素可接的雨水体积。由于有可能存在，当前元素比栈顶元素大，但比栈顶元素的左元素小的情况，也有可能当前元素比栈顶元素的左元素大。然后，可接的个数应该取“最短的那个木板”。因此，寻找二者中的最小高度作为与栈顶元素的高度差，相减求出可接雨水高度。然后，再找出长度与之相乘，就可得到可接的雨水体积，也就是可接雨水个数。

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size(), res = 0;
        stack<int> st;
        st.push(0);
        for(int i = 1; i < n; i++) {           
            while(!st.empty() && height[st.top()] < height[i]) {
                int mid = st.top();     st.pop();
                if(!st.empty()) {
                	// 找到高度边界中的最小值（最短的那块木板） - 本次高度
                    int h = min(height[st.top()], height[i]) - height[mid];
                    // 右凹槽下标 - 左凹槽下标 - 1 （因为只需要到中间凹槽，会少减去一个1）
                    int w = i - st.top() - 1; 
                    res += h * w;                    
                }
            }
            // 当为单调不增时或弹栈后都入栈当当前情况，用于后续对比
            st.push(i);
        }

        return res;
    }
};

参考文章：42. 接雨水