【leetcode】503. 下一个更大元素 II（next-greater-element-ii）（栈）[中等]

链接

https://leetcode-cn.com/problems/next-greater-element-ii/

耗时

解题：15 min
题解：9 min

题意

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

注意： 输入数组的长度不会超过 10000。

思路

第一次遍历数组中每个元素，将其放入堆栈，在入栈之前，判断，如果当前元素大于栈顶元素，则当前元素即是栈顶元素的下一个更大的数，将结果数组中栈顶元素对应的位置变为当前元素，并出栈栈顶元素，直到栈顶元素不再大于当前元素，入栈当前元素。

因为是循环数组，所以需要第二次遍历数组中每个元素，这一次不再入栈，其余与第一次相同。

每个元素入栈出栈一次。

时间复杂度： O ( n ) O(n) O(n)

AC代码

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n = nums.size();
        vector<int> ans(n, -1);
        stack<int> st;
        for(int i = 0; i < n; ++i) {
            while(!st.empty() && nums[i] > nums[st.top()]) {
                ans[st.top()] = nums[i];
                st.pop();
            }
            st.push(i);
        }
        for(int i = 0; i < n; ++i) {
            while(!st.empty() && nums[i] > nums[st.top()]) {
                ans[st.top()] = nums[i];
                st.pop();
            }
        }
        return ans;
    }
};