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;

    }
    
};
相关推荐
一起努力啊~3 分钟前
算法刷题--哈希表
算法·面试·散列表
willingli9 分钟前
c语言经典100题 61-70题
c语言·开发语言·算法
我是小疯子6613 分钟前
深入解析C++右值引用与移动语义
java·开发语言·算法
踩坑记录15 分钟前
leetcode hot100 56.合并区间 medium
leetcode
源代码•宸23 分钟前
Golang原理剖析(Map 源码梳理)
经验分享·后端·算法·leetcode·golang·map
程序员-King.25 分钟前
day132—链表—K个一组翻转链表(LeetCode-25)
leetcode·链表·贪心算法
Narrastory29 分钟前
手把手实现蚁群算法:从数学原理到代码实践
算法
mit6.82439 分钟前
八皇后变题hash|网格dp
算法
bybitq1 小时前
LeetCode-437-路径总和3
算法
im_AMBER1 小时前
数据结构 18 【复习】广义表 | 各种内部排序 | 二叉排序树的平均查找长度 ASL
数据结构·笔记·学习·排序算法