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;

    }
    
};
相关推荐
客梦37 分钟前
数据结构-栈与队列
数据结构·笔记
chao18984438 分钟前
MATLAB 实现声纹识别特征提取
人工智能·算法·matlab
zhishidi41 分钟前
推荐算法之:GBDT、GBDT LR、XGBoost详细解读与案例实现
人工智能·算法·推荐算法
货拉拉技术42 分钟前
货拉拉RAG优化实践:从原始数据到高质量知识库
数据库·算法
AKDreamer_HeXY1 小时前
ABC434E 题解
c++·算法·图论·atcoder
罗湖老棍子1 小时前
完全背包 vs 多重背包的优化逻辑
c++·算法·动态规划·背包
TL滕1 小时前
从0开始学算法——第四天(题目参考答案)
数据结构·笔记·python·学习·算法
potato_may1 小时前
C++ 发展简史与核心语法入门
开发语言·c++·算法
Liangwei Lin1 小时前
洛谷 P1443 马的遍历
数据结构·算法