力扣 面试题 17.21. 直方图的水量 dp/双指针/单调栈

https://leetcode-cn.com/problems/volume-of-histogram-lcci/

思路一：单独考虑每一个位置$i$，假设在$[0,i-1]$内的最大值为$maxl$，在$[i+1,n-1]$内的最大值为$maxr$，则位置$i$的存水量等于$max(height[i],min(maxl,maxr))-height[i]$。那么遍历两边即可求出结果。

class Solution {
public:
    int trap(vector<int>& height) {
        int siz=height.size();
        if(!siz)
            return 0;
        vector<int> maxarr(siz);
        int MAX=0;
        for(int i=0;i<siz;i++){
            maxarr[i]=MAX;
            MAX=max(MAX,height[i]);
        }
        int ans=0,tmp;
        MAX=height[siz-1];
        for(int i=siz-2;i>=0;i--){
            tmp=min(MAX,maxarr[i]);
            if(tmp>height[i])
                ans+=tmp-height[i];
            MAX=max(MAX,height[i]);
        }
        return ans;
    }
};

思路二：双指针。这个思路相当于思路一的反向思维。思路一是对于每个位置，确定左边界和右边界，然后计算贡献。思路二是先确定左右边界，再统计位置的贡献。搞两个指针，$l$从左向右扫，$r$从右向左扫，不妨假设$height[l]<=height[r]$，那么对于从$l$开始的单调递减区间内的每一个值$v$，对答案都有$height[l]-v$的贡献；如果$height[l]>height[r]$，那么类似的从$r$开始想左扫就好了。如果不存在这样的区间的话，说明这个位置不能产生贡献，移动指针选定下一对左右边界即可。

class Solution {
public:
    int trap(vector<int>& height) {
        int siz=height.size();
        int l=0,r=siz-1,ans=0;
        while(l<r){
            if(height[l]<=height[r]){
                int tmp=height[l];
                while(++l<r&&height[l]<tmp)
                    ans+=tmp-height[l];
            }
            else{
                int tmp=height[r];
                while(--r>l&&height[r]<tmp)
                    ans+=tmp-height[r];
            }
        }
        return ans;
    }
};

思路三：单调栈，比较难想到。和思路二有一点点像。想象一下，我们维护了一个单调递减的栈，假设当前右侧边界高度为$h_1$，栈顶高度为$h_2$，栈顶的下一个元素的高度为$h_3$，那么只要$h_1>h_2$，就可以以$h_2$为底，$h_1$为右边界，$h_3$为左边界，把中间这一块填满。详见代码，可以自己画图理解一下。

class Solution {
public:
    int trap(vector<int>& height) {
        //单调栈 但是存储的是下标
        stack<int> s;
        int ans=0,siz=height.size();
        for(int i=0;i<siz;i++){
            int tmp;
            while(!s.empty()&&height[i]>=height[tmp=s.top()]){
                s.pop();
                if(!s.empty()){
                    //s.top为左边界 tmp为底 i为右边界
                    //MIN-height[tmp] 为填充的区块的高度
                    //i-s.top-1 为填充的区块的宽度
                    int MIN=min(height[s.top()],height[i]);
                    ans+=(MIN-height[tmp])*(i-s.top()-1);
                }
            }
            s.push(i);
        }
        return ans;
    }
};