单调栈复健,顺便,牺牲一下吧,空间复杂度!一切献给AC

Problem: 503. 下一个更大元素 II

  1. 思路
  2. 解题方法
  3. 复杂度
  4. Code

思路

单调栈走起,维护一个单调递减栈。

解题方法

牺牲一下空间复杂度,开辟一个2*n的数组,遍历此数组。这样对于前置数组的末尾那几位,就可以通过继续遍历后置数组,从而能够正确获取到最近的比它要大的数。并通过一个max来记录数组中最大的值

举个例子:假设nums是[1,2,3,2,1]

则2*n的newNums就是[1,2,3,2,1,1,2,3,2,1]

当我们遍历到前置数组 的最后一项 ,1时

此时的单调栈如下:

[2,3,4]

注意哦,单调栈中保存的是数组下标index哦!!!

也就是说,此时的结果数组为:[2,3,待定,待定,待定,待定,待定,待定,待定,待定]

结果数组也是2*n

只遍历一次显然不够,于是继续遍历后置数组

[1,2,3,2,1]

逻辑同第一次遍历,有区别的是,在第二次遍历假如当前项和最大值相同,则要将对应的结果数组 中的index置为-1即可。

res[i - nums.length] = -1

最后我们得到这样的结果数组 :

[ 2, 3, -1, 3, 2, 2, 3, 0, 0, 0 ]

返回前n项即可:

[ 2, 3, -1, 3, 2]

复杂度

时间复杂度:

O(n)

空间复杂度:

O(n)

Code

js 复制代码
/**
 * @param {number[]} nums
 * @return {number[]}
 */
var nextGreaterElements = function (nums) {
    let stk = [];
    let res = Array(nums.length * 2).fill(0);
    let newNums = [...nums, ...nums]
    let max = -Infinity
    for (let i = 0; i < newNums.length; i++) {
        while (stk.length > 0 && newNums[stk[stk.length - 1]] < newNums[i]) {
            res[stk.pop()] = newNums[i]
        }
        if (max < newNums[i]) {
            max = newNums[i]
        }
        if (newNums[i] === max && i >= nums.length) {
            res[i - nums.length] = -1;
        }
        stk.push(i)
    }

    return res.slice(0, nums.length)
};
相关推荐
薛定e的猫咪4 分钟前
(ICLR 2026)DF-META:面向 FJSP 的无导数元学习实例自适应调度
人工智能·深度学习·学习·算法·机器学习
java_nnnn1 小时前
LeetCode刷题滑动窗口(水果成篮+找到字符串中所有字母的异位词)Java
java·开发语言·leetcode
小羊没烦恼!2 小时前
异步编程:线程概述及使用
java·开发语言·mysql·算法·c#
wanderist.2 小时前
202609CSP认证P3
算法
云淡风轻~窗明几净2 小时前
角谷猜想最接近严格证明的测试 Version1.1.1
人工智能·算法·机器学习·图论
机器学习之心3 小时前
PINN × SVM × NSGA-II × 熵权TOPSIS:一套跑通「建模—优化—决策」的多目标优化框架
算法·机器学习·支持向量机
别动我齐刘海3 小时前
机器人 C++ / 系统联调补强2——异常处理、Protobuf
linux·c++·人工智能·学习·算法·机器学习·机器人
文丑颜不良啊4 小时前
LeetCode276~300题目解析
java·leetcode
荆棘鸟智能4 小时前
短历时洪水预报算法实践:水文模型与 LightGBM 残差订正融合
人工智能·算法·机器人
葫三生5 小时前
关于三生原理系列被技术开发社区批量收录或引发策展边界讨论的一种思考
大数据·人工智能·深度学习·算法·机器学习·数据挖掘·开源