503 下一个更大元素 II(单调栈、破环成链技巧)

1. 问题描述:

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

示例 1:

输入: [1,2,1]
输出: [2,-1,2]

解释: 第一个 1 的下一个更大的数是 2;
数字 2 找不到下一个更大的数; 
第二个 1 的下一个最大的数需要循环搜索,结果也是 2。
注意: 输入数组的长度不会超过 10000。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/next-greater-element-ii

2. 思路分析:

这道题目类似于之前力扣的496题,也是寻找下一个更大的数字,但是这里的数组可以看成是循环的,环形数组处理起来就比较麻烦,这里可以使用一个技巧是将原数组复制一遍接到原数组的后面,这样就可以将环形数组变为普通的一维数组,可以发现每一个数字后面都可以覆盖环形数组的长度为n的路径所以是符合要求的,并且求解出的答案也是满足要求的(例如例子[1,2,1][1,2,1]),这样我们就可以直接使用单调栈的模板了,这里需要注意的是只有在下标在n之内才更新答案(因为更新的才是原数组对应的答案数组)。

3. 代码如下:

from typing import List


class Solution:
    def nextGreaterElements(self, nums: List[int]) -> List[int]:
        n = len(nums)
        # 将原数组复制一份接到后面
        nums = nums + nums[:]
        # 栈中存储的是元素下标
        stack = list()
        res = [-1] * n
        for i in range(2 * n):
            while stack and nums[stack[-1]] < nums[i]:
                # 下标在n之内才更新答案
                if stack[-1] < n:
                    res[stack[-1]] = nums[i]
                stack.pop()
            stack.append(i)
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值