LeetCode-Python-503. 下一个更大元素 II

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

示例 1:

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

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

思路：

首先判断题型：

本题需要后面的数来决定前面数的答案，所以自然地就想到要用栈，stack来解题。

首先处理corner case，

开辟一个栈，栈里存放的是数组元素的下标

然后线性扫描整个数组，如果栈是空的，或者栈顶元素对应的数组元素比当前元素大，那么就把当前元素的下标压入栈，

 否则，就说明栈顶元素对应的数组元素的下一个更大元素就是当前元素，然后pop栈顶元素，重复这个过程直到栈顶元素对应的数组元素》=当前元素或者栈为空为止。

一趟线性扫描结束之后，如果栈不为空，就说明还要扫一次找到需要循环查找的结果，

就从头再扫一次，思路跟上面一样。

class Solution(object):
    def nextGreaterElements(self, nums):
        """
        :type nums: List[int]
        :rtype: List[int]
        """
        if not nums:
            return nums
        max_element = max(nums)
        stack = list()
        res = [-1 for i in range(len(nums))]
        
        for i, x in enumerate(nums):
                
            if not stack or nums[stack[-1]] >= x:
                stack.append(i)
            else:
                # print stack, res
                while(stack and nums[stack[-1]] < x):
                      res[stack[-1]] = x
                      stack.pop()
                if x != max_element:#最大的不用放进栈，直接默认-1就好
                    stack.append(i)
                    
        # print stack, res            
        if stack: #还有需要循环搜索的元素
            for i, x in enumerate(nums):
                if not stack:
                      break
                while stack and x > nums[stack[-1]]:
                      res[stack[-1]] = x
                      stack.pop()
                     
        return res

举一反三，本题跟以下题是同类题：

LeetCode-Python-503. 下一个更大元素 II

LeetCode-Python-739. 每日温度

LeetCode-Python-496. 下一个更大元素 I

LeetCode-Python-1030. 链表中的下一个更大节点