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

503.下一个更大元素II

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

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

class Solution {
public:
    vector nextGreaterElements(vector& nums) {
        vectornums1(nums.begin(),nums.end());
        nums.insert(nums.end(),nums1.begin(),nums1.end());
        for(int i=0;ires(nums.size(),-1);
        if(nums.size()==0)return res;
        stackst;
        st.push(0);
        for(int i=1;inums[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& height) {
        stackst;
        st.push(0);
        int sum=0;
        for(int i=1;iheight[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;
    }
};