代码随想录算法训练营第六十二天| LeetCode503.下一个更大元素II、42. 接雨水

一、LeetCode503.下一个更大元素II

题目链接/文章讲解/视频讲解：https://programmercarl.com/0503.%E4%B8%8B%E4%B8%80%E4%B8%AA%E6%9B%B4%E5%A4%A7%E5%85%83%E7%B4%A0II.html

状态：已解决

1.思路 

        这道题和前面的题有稍许不同，考察了循环数组的运用，而一般循环数组的题我们采取的措施是将数组复制一份与原数组拼起来，结果数组也要跟着延长（因为在遇到当前元素比栈顶元素大的情况，出栈栈顶元素的时候要将当前元素的值作为栈顶元素对应的结果，如果不延长我们还要判断一下栈顶元素是否在0~nums.size(）的范围内），这样求解出来的result数组就是原数组的两倍，因此最后还需要重塑一下数组为原数组的大小（即取结果数组前面的部分即可）

2.代码实现

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> result(nums.size()*2,-1);
        vector<int> nums1(nums.begin(),nums.end());
        nums.insert(nums.end(),nums1.begin(),nums1.end());
        stack<int> s;
        if(nums.size()==0) return result;
        s.push(0);
        for(int i=1;i<nums.size();i++){
            if(nums[i]<=nums[s.top()]){
                s.push(i);
            }else{
                while(!s.empty() && nums[i] > nums[s.top()]){
                    result[s.top()] = nums[i];
                    s.pop();
                }
                s.push(i);
            }
        }
        result.resize(nums.size()/2);
        return result;
    }
};

二、42. 接雨水

题目链接/文章讲解/视频讲解：https://programmercarl.com/0042.%E6%8E%A5%E9%9B%A8%E6%B0%B4.html

状态：已解决 

1.思路

        感觉我讲不好，建议还是看连接中的文章讲解。这道题理解的关键在于如何计算积水量：根据面积算的。

        如图，我们将蓝色部分拆成上下拼接的方形，然后计算各个方形的面积。我们知道，计算面积实则就是计算长*宽。 这里的长就是围成凹形区域的两个柱子之间的距离，宽就是两个柱子的最小值减去底部柱子的高度。那么问题又来到如何找到围城凹形区域的两根柱子上来了：其实就是维护单调栈（从栈顶到栈底）单增的顺序。当当前遍历元素高度大于栈顶时，说明当前元素可以和栈里面的部分元素构成凹形区域：

        那么，栈顶元素就是底部柱子，栈顶元素的下一个元素就凹形区域的左柱子，当前元素就是凹形区域的右柱子。

        理解了上面部分，99%的思路就能想明白了，但还有一点，就是当当前元素等于栈顶元素高度时，如图。

 

        此时凹形区域应该由右边柱子围成，故遇到相等情况时，弹出栈顶元素，把当前元素放进去，替换掉上一个柱子。

2.代码实现 

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int> s;
        int sum=0;
        s.push(0);
        for(int i=1;i<height.size();i++){
            if(height[i]<height[s.top()]){
                s.push(i);
            }else if(height[i] == height[s.top()]){//注意这个相等情况。
                s.pop();
                s.push(i);
            }else{
                while(!s.empty() && height[i]>height[s.top()]){
//注意，由于凹形区域可能由几个小的凹形区域构成，因此计算完一块凹形区域后，
//还要看当前元素是否又跟新的栈顶构成凹形区域，故用while
                    int bottom = height[s.top()];//底部柱子
                    s.pop();
                    if(!s.empty()){//注意要看栈弹出一个元素是否为空
                        int leftHeight = height[s.top()];//左柱子
                        int rightHeight = height[i];//右柱子
                        sum += (min(rightHeight,leftHeight)-bottom)*(i-s.top()-1);//长*宽
                    }
                }
                s.push(i);
            }
        }
        return sum;
    }
};