ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

LeetCode/下一个更大元素

2022-07-03 04:00:09  阅读:167  来源: 互联网

标签:nums int 元素 nums1 vector 更大 LeetCode nums2


nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
给你两个 没有重复元素 的数组 nums1 和 nums2 ,下标从 0 开始计数,其中nums1 是 nums2 的子集。
对于每个 0 <= i < nums1.length ,找出满足 nums1[i] == nums2[j] 的下标 j ,并且在 nums2 确定 nums2[j] 的 下一个更大元素 。如果不存在下一个更大元素,那么本次查询的答案是 -1 。
返回一个长度为 nums1.length 的数组 ans 作为答案,满足 ans[i] 是如上所述的 下一个更大元素

1. 单调栈+哈希表

这是一道简单题,直接暴力解法时间复杂度为O(nm),即对于num1中每个元素,先在nums2中找到,然后往后寻找第一个更大值
但由于num1是num2的子集,以及不存在重复元素
可以直接对num2先进行评估,记录每个值右侧最大值是什么,然后再遍历num1即可,时间复杂度为O(m+n)
这里使用单调栈得到右侧最大值,以及通过哈希表进行记录

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> m;
        //记录某个值右侧最大值,单调栈
        stack<int> s;
        for(int i=0;i<nums2.size();i++){
            //为空或者小于栈底
            if(s.empty()||s.top()>nums2[i]){ s.push(nums2[i]); continue;}
            while(!s.empty()&&nums2[i]>s.top()){
                m[s.top()] = nums2[i];//建立哈希关系
                s.pop();//出栈
            }
            s.push(nums2[i]);//元素入栈
        }
        //单调递减栈构建结束
        while(!s.empty()){
            m[s.top()] = -1;//右侧无更小值
            s.pop();
        }
        vector<int> res(nums1.size());
        for(int i=0;i<nums1.size();i++)
            res[i] = m[nums1[i]];
        return res;
    }
};

2. 循环有重复数组

通过取余运算来遍历循环数组,同样利用单调栈,获得对应结果

class Solution {
public:
    vector<int> nextGreaterElements(vector<int>& nums) {
        vector<int> res(nums.size(),-1);//初始值给-1,最后更新完不在单调栈中的其他值即可
        //记录某个值右侧最大值,单调栈
        stack<int> s;
        int n = nums.size();
        for(int i=0;i<n*2;i++){//遍历2n-1个数即可
            //为空或者小于栈底
            while(!s.empty()&&nums[i%n]>nums[s.top()]){
                res[s.top()] = nums[i%n];//建立哈希关系
                s.pop();//出栈
            }
            s.push(i%n);//下标元素入栈
        }
        return res;
    }
};

3. 下一个更大元素III

给你一个正整数 n ,请你找出符合条件的最小整数,其由重新排列 n 中存在的每位数字组成,并且其值大于 n
如果不存在这样的正整数,则返回 -1

类似下一个排列题目

class Solution {
public:
    int nextGreaterElement(int n) {
        auto&& s = to_string(n);
        return !next_permutation(s.begin(), s.end()) || stol(s) > INT_MAX ? -1 : stoi(s);
    }
};

完整解法

class Solution {
public:
    int nextGreaterElement(int n) {
        auto nums = to_string(n);//转换成字符串
        int i =  nums.length() - 2;//防止溢出
        while (i >= 0 && nums[i] >= nums[i + 1]) //从后往前找第一个严格升序对
            i--;
        if (i < 0) return -1;//不存在升序对,已经是最大值

        int j = nums.size() - 1;
        while (j >= 0 && nums[i] >= nums[j])//从后往前找第一个严格比i大的值
            j--;

        swap(nums[i], nums[j]);//替换对应值
        reverse(nums.begin() + i + 1, nums.end());//同时把后面的降序转换为升序
        long ans = stol(nums);//重新转换成数
        return ans > INT_MAX ? -1 : ans;
    }
};

标签:nums,int,元素,nums1,vector,更大,LeetCode,nums2
来源: https://www.cnblogs.com/929code/p/16439143.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有