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

503.下一个更大元素II

题目描述：给定一个循环数组（最后一个元素的下一个元素是数组的第一个元素），输出每个元素的下一个更大元素。数字 x 的下一个更大的元素是按数组遍历顺序，这个数字之后的第一个比它更大的数，这意味着你应该循环地搜索它的下一个更大的数。如果不存在，则输出 -1。

思路：数组扩大一倍，这样就可在最大节省空间和算力的效果下实现循环遍历。

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

42. 接雨水

题目描述：给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

class Solution {
public:
    int trap(vector<int>& height) {
        stack<int>st;
        st.push(0);
        int sum=0;
        for(int i=1;i<height.size();i++)
        {
            while(!st.empty()&&height[i]>height[st.top()])
            {
                int mid = st.top();
                st.pop();
                if(!st.empty())
                {
                    int h = min(height[st.top()],height[i])-height[mid];
                    int w=i-st.top()-1;
                    sum+=h*w;
                }
            }
            st.push(i);
        }
        return sum;
    }
};