下一个更大元素II(python)

题目描述：

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

示例1：

        输入：[1,2,1]

        输出：[2,-1,2]

        解释：第一个1的下一个更大的数是2；

        数字2找不到下一个更大的数；

        第二个1的下一个最大的数需要循环搜索，结果也是2。

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

class Solution(object):
    def nextGreaterElements(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        #思路：用list作为栈，将nums中的数据依次压入，边压入边进行对比
        if len(nums) == 0:
            return []
        
        stack,res = [0],[0]*len(nums)*2
        tmp = nums * 2
        
        for i in range(1,len(tmp)):
            while stack and tmp[stack[-1]] < tmp[i]:
                res[stack[-1]] = tmp[i]
                stack.pop()
            stack.append(i)
            
        for i in stack:
            res[i] = -1
            
        return res[:len(nums)]

 菜鸟一枚，代码仅供参考，如有问题，望指正~