【Leetcode单调栈】503. 下一个更大元素 II(有以后都这么写的单调栈的更简洁写法!!代码可复用!!)

Leetcode503

1.问题描述

在这里插入图片描述

2.解决方案

1.这道题和【Leetcode单调栈】739. 每日温度(有关于单调栈类型题的深入理解概括!!)也几乎如出一辙。不同的时候本题要循环数组了。
2.相信不少同学看到这道题,就想那我直接把两个数组拼接在一起,然后使用单调栈求下一个最大值不就行了!确实可以!将两个nums数组拼接在一起,使用单调栈计算出每一个元素的下一个最大值,最后再把结果集即result数组resize到原数组大小就可以了。
3.其实也可以不扩充nums,而是在遍历的过程中模拟走了两边nums。


解法一:繁琐写法
class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        //1.准备递减栈
        stack<int> stack;
        vector<int> ans(nums.size(),-1);
        if(nums.size()<2) return ans;

        //2.遍历构建单调栈
        stack.push(0);
        for(int j=1;j<=2*nums.size()-1;j++){
            //遍历两遍数组
            int i=j%nums.size();
            //情况1,2
            if(nums[stack.top()]>=nums[i]){
                stack.push(i);
                continue;
            }
            //情况3:出结果的一种情况
            if(nums[stack.top()]<nums[i]){
                while(stack.empty()== false&&nums[stack.top()]<nums[i]){
                    ans[stack.top()]=nums[i];
                    stack.pop();
                }
                stack.push(i);
                continue;
            }
        }
        return ans;
    }
};


解法二:单调栈简洁写法(以后都这么写!!)
class Solution1{
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        //1.准备阶段
        stack<int> st;
        vector<int> result(nums.size(), -1);
        if(nums.size()==0) return result;

        //2.遍历构造单调栈阶段
        for(int i=0;i<nums.size()*2;i++){
            //模拟遍历两边nums,注意一下都是用i % nums.size()来操作
            //情况1,2,3->合一起写了很简洁!
            while(st.empty()== false&&nums[st.top()]<nums[i%nums.size()]){
                result[st.top()]=nums[i % nums.size()];
                st.pop();
            }
            st.push(i % nums.size());
        }
        return result;
    }
};
  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值