面试题 17.21. 直方图的水量

接雨水原题，hard+=1。
考虑每个柱子上面能接多少是最重要的，被中间两个1的柱子迷惑住了，完全不用迷惑，因为1-1就是0，那么不用在乎它的宽度了。

class Solution {
public:
    int trap(vector<int>& height) {
        //不就是接雨水吗，计算每个格子上方可以接多少雨水就完事了
        //用单调栈，因为碰到左右都比自己的高的时候才能接
        //用左右低的那个-中间的高度然后*宽度就行了
        if(height.size() <= 2) return 0;
        stack<int> sta;//存的是下标，方便计算宽度
        int res = 0;
        for(int i = 0; i < height.size(); ++i){
            //先判断栈是否为空，不然取栈顶会出错
            while(!sta.empty() && height[sta.top()] < height[i]){
                int mid = sta.top();
                sta.pop();
                //还需要来个判断，如果左边没有接雨水的柱子，直接退出
                if(sta.empty()) break;
                int gao = min(height[sta.top()],height[i])-height[mid];
                int kuan = i-sta.top()-1;//只求中间的宽度
                res += gao*kuan;
            }
            sta.push(i);
        }
        return res;
    }
};