503. 下一个更大元素 II

给定一个循环数组 numsnums[nums.length - 1] 的下一个元素是 nums[0] ),返回 nums 中每个元素的 下一个更大元素

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

示例 1:

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

单调栈

写过 下一个更大元素I 的同学,应该可以注意到,这道与I的区别仅仅是这里是一个循环数组。也就是说,单调栈遍历一遍后并不能 遍历完全部,应当遍历两遍,如第一遍遍历到头则继续向前遍历。

这里可以使用索引取余的方式来实现逻辑上的遍历两次:同时使用单调栈存索引而并非存值,方便存答案,避免重复元素覆盖问题。

根据上图,当第一次遍历完时,1、2、3、都已确定下一个最大。末尾3还不能确定。通过索引方式可以模拟继续向前遍历,则遍历到前段的那个3,根据单调栈回退,删去1、2、3索引,找到下一个最大元素4

cpp 复制代码
class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        
        int n = nums.size();
        vector<int> res(n, -1);
        vector<int> st; // 单调栈:存索引;从栈顶到栈底依次递增。

        for (int i = 2*n - 1; i >= 0; --i) {
            int idx = i % n;
            //由于单调栈用于存索引,因此nums[st.back()]为栈中实际元素
            while (!st.empty() && nums[st.back()] <= nums[idx]) {
                st.pop_back();
            }
            //覆盖
            if (!st.empty()) {
                res[idx] = nums[st.back()];
            }

            st.push_back(idx);
        }

        return res;

    }
    
};
相关推荐
源代码•宸2 小时前
Leetcode—620. 有趣的电影&&Q3. 有趣的电影【简单】
数据库·后端·mysql·算法·leetcode·职场和发展
2301_800256112 小时前
地理空间数据库中的CPU 和 I/O 开销
数据库·算法·oracle
一个不知名程序员www3 小时前
算法学习入门---结构体和类(C++)
c++·算法
XFF不秃头5 小时前
力扣刷题笔记-旋转图像
c++·笔记·算法·leetcode
王老师青少年编程6 小时前
csp信奥赛C++标准模板库STL案例应用3
c++·算法·stl·csp·信奥赛·lower_bound·标准模版库
有为少年6 小时前
Welford 算法 | 优雅地计算海量数据的均值与方差
人工智能·深度学习·神经网络·学习·算法·机器学习·均值算法
Ven%7 小时前
从单轮问答到连贯对话:RAG多轮对话技术详解
人工智能·python·深度学习·神经网络·算法
山楂树の7 小时前
爬楼梯(动态规划)
算法·动态规划
谈笑也风生7 小时前
经典算法题型之复数乘法(二)
开发语言·python·算法