【LeetCode】503. 下一个更大元素 II 解题报告 (python)

原题地址：https://leetcode-cn.com/problems/next-greater-element-ii/

题目描述：

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

示例 1:

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

 

解题方案：

栈，这里有个小技巧，对数组double一下：

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        len_ = len(nums)
        res = [-1 for _ in range(len_)]
        sta = []
        
        double = nums + nums
        for idx, num in enumerate(double):
            while sta and nums[sta[-1]] < num:
                res[sta[-1]] = num
                sta.pop()
            if idx < len_:
                sta.append(idx)
        return res