LeetCode 503 下一个更大元素2

LeetCode 503 下一个更大元素2

1.题目描述：

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

示例 1:

输入: [1,2,1]
输出: [2,-1,2]
解释: 第一个 1 的下一个更大的数是 2；
数字 2 找不到下一个更大的数； 
第二个 1 的下一个最大的数需要循环搜索，结果也是 2。

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

2.题解：

1. 循环数组找下一个更大的值，可以用取模的方式，或者将原数组后面复制一份原数组
2. 单调栈思路：建立一个单调递减的栈，栈中储存元素对应的index，如果新入栈元素比栈顶元素大，栈顶出栈，直到不比栈顶元素大为止（或栈为空为止）。在出栈过程中，就已经确定右边第一个比其本身大的数，并存入数组ans中。

时间复杂度：O(n)，n为序列长度。数组中的元素最多被遍历2次，每个元素入栈，出栈最多只有四次。

空间复杂度：O(n)，n为序列长度，空间大小主要取决于栈的大小，栈的大小的至多为2n-1。

C++：

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        int n = nums.size() ;
        stack<int> st ;
        vector<int> ans(n ,-1) ; //所有位置初始化为-1
        for ( int i = 0 ; i < 2*n-1; i++) {
            while(!st.empty()&& nums[i%n] > nums[st.top()]) {
                ans[st.top()] = nums[i%n] ; //采用取模方式
                st.pop() ;
            }
            st.push(i%n) ;
        }
        return ans ; 
    }
};

Golang：

func nextGreaterElements(nums []int) []int {
    n := len(nums) 
    st := make([]int, n) 
    ans := make([]int, n)
  //初始化为-1
    for i:= range ans { 
        ans[i] = -1
    }
    for i:= 0; i < 2*n -1; i++ {
        for len(st)>1 && nums[i%n] > nums[st[len(st)-1]] {
            ans[st[len(st)-1]] = nums[i%n] 
            st = st[:len(st)-1]
        }
        st = append(st, i%n)
    } 
    return ans
}

原题：下一个更大元素