leetcode503. 下一个更大元素Ⅱ（单调栈解法详解）

一、注

如果您还不知道什么是单调栈；如果你想要了解leetcode中单调栈题目的汇总，可以看博主的这篇博客。

       何谓单调栈？leetcode单调栈题目汇总

 

二、原题

原题链接  503. 下一个更大元素Ⅱ

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

来源：力扣（LeetCode）
 

三、解析

这道题目是 leetcode 496.下一个更大元素（单调栈解法） 的进阶版，二者的的不同点在于：一是503的数组中可能有重复元素，而496的数组不包含重复元素，因此503不能采用哈希表的形式；二是503中是一个循环数组。注意，这道题目中是要返回下一个更大的元素的值，因此代码中另next_bigger存储该值。鉴于是处理一个循环数组，那我们只需要遍历两遍nums数组就可以了。以[1,4,3,2]为例，我们先初始化next_bigger为[-1，-1，-1，-1]，-1表示该元素后面没有比它更大的元素；构造空单减栈min_stack，栈中存储下标。

      创建变量length记录数组nums的长度。

      i=0时，nums[i % length]=1，此时为空栈，直接将下标0进栈；

      i=1时，nums[i % length]=4，栈顶对应元素1小于4，将下标0出栈、下标1进栈，同时知元素1后面比他大的元素是4，故修改  next_bigger[0] = 1；

      i=2时，nums[i % length]=3，栈顶对应元素4大于3，直接将下标2进栈；

      i=3时，nums[i % length]=2，栈顶对应元素3大于2，直接将下标3进栈；

      如果不是循环数组，那么这道题到此就结束了。

      i=4时，nums[i % length ]=1，栈顶对应元素2大于1，将下标0进栈；

      i=5时，nums[i % length ]=4，栈顶对应元素1小于4，将下标0出栈，修改next_bigger[0]=i % length = 1;此时新的栈顶对应元素2小于4，将下标3出栈，修改next_bigger[3] =  i % length = 1；同理，修改 next_bigger[2] =  i % length = 1；这时栈中剩余 1和1（下标值，对应元素值4）。

     i = 6时，栈顶对应元素4大于3，将下标2进栈；

     i = 7时，栈顶对应元素3大于2，将下标3进栈。

     循环结束，返回next_bigger即可。

class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        length = len(nums)
        next_bigger = [-1] * length
        min_stack = [] # 存储下标
        for i in range(2*length):
            while(min_stack and nums[min_stack[-1]] < nums[i % length]):
                top = min_stack.pop()
                next_bigger[top] = nums[i % length]
            min_stack.append(i % length)
        return next_bigger