力扣leetcode面试题 17.21. 直方图的水量

面试题 17.21. 直方图的水量
给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 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
通过次数33,666提交次数53,537

思路一：通过用辅助数组的思想来寻找某个点左边最大和右边最大；
寻找到最大中最小的值减去height[i]就是存水量。

创建两个长度为 n的数组left[n],right[n];
分别从左右两边遍历数组，
(1)—>left存储max(left[i-1],left[i])
(2)—>right存储min(right[i],right[i-1])
(3)—>计算实际存水量min(left[i],right[i])-height[i];

备注：图片来源于leetcode

class Solution{
public:
    int trap(vector<int>& height) {
        int n=height.size();
        if(n=0)
            return 0;
        vector<int>left(n);
        vector<int>right(n);//通过用辅助数组的思想来寻找某个点左边最大和右边最大；
        left[0]=height[0];
        for(int i=1;i<n;i++)//(1)--->left存储max(left[i-1],left[i])
        {
            left[i]=max(left[i],left[i-1]);

        }
        right[n-1]=height[n-1];
        for(int i=n-2;i>=0;i--)//(2)--->right存储min(right[i],right[i-1])
        {
            right[i]=max(right[i+1],right[i]);
        }

        int sum=0;
        for(int i=0;i<n;i++)//(3)--->计算实际存水量min(left[i],right[i])-height[i];
        {
            sum+=min(left[i],right[i])-height[i];
        }//寻找到最大中最小的值减去height[i]就是存水量。

        return sum;
    }
};

方法二：单调栈
刚开始接触单调栈，有忆点不熟悉

public:
    int trap(vector<int>& height)
    {
        int ans=0;
        stack<int >st;
        int n=height.size();
        for(int i=0;i<n;i++)
        {
           while(!st.empty()&&height[i]>height[st.top()])
           //如果最新的数组元素大小大于栈顶元素的大小，说明此时可以形成类似容器的东西
           //此时要记录容器的右边就是i,容器的左边就是left=st.top();
           //(i-left-1)容器的宽
           //(min(height[i],height[left])-height[top]);容器的高
           {
               int top=st.top();
               st.pop();
               if(st.empty())
               {break;
               }
               int left=st.top();
               ans+=(i-left-1)*(min(height[i],height[left])-height[top]);
           }
           st.push(i);//每次都要把数组坐标压入栈
        }
return 0;

    }
};