zoukankan      html  css  js  c++  java
  • 下一个更大元素

    496. 下一个更大元素 I

    给你两个 没有重复元素 的数组 nums1 和 nums2 ,其中nums1 是 nums2 的子集。请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在,对应位置输出 -1 。

    示例 :

    输入: nums1 = [4,1,2], nums2 = [1,3,4,2].

    输出: [-1,3,-1]

    解释:

        对于 num1 中的数字 4 ,你无法在第二个数组中找到下一个更大的数字,因此输出 -1 。

        对于 num1 中的数字 1 ,第二个数组中数字1右边的下一个较大数字是 3 。

        对于 num1 中的数字 2 ,第二个数组中没有下一个更大的数字,因此输出 -1 。

    class Solution {
    public:
        vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) 
        {
            unordered_map<int, int> m;
            stack<int> s; //单调栈
            for(size_t i = 0; i < nums2.size(); i++){
                while(!s.empty() && s.top() < nums2[i]){
              //当前值比栈顶值大,则栈顶值右边的第一个比其大的数就是当前值,将栈顶值出栈      
                    m[s.top()] = nums2[i];  
                    s.pop();
                }
                s.push(nums2[i]);
            }
            while(!s.empty()){//栈中剩下的数表明在其右边没有比其更大的数
                m[s.top()] = -1;
                s.pop();
            }
            vector<int> ret;
            for(int each : nums1){
                ret.push_back(m[each]);
            }
            return ret;
        }
    };
    

    503. 下一个更大元素 II

    // 版本一
    class Solution {
    public:
        vector<int> nextGreaterElements(vector<int>& nums) {
            // 拼接一个新的nums
            vector<int> nums1(nums.begin(), nums.end());
            nums.insert(nums.end(), nums1.begin(), nums1.end());
            // 用新的nums大小来初始化result
            vector<int> result(nums.size(), -1);
            if (nums.size() == 0) return result;
    
            // 开始单调栈
            stack<int> st;
            for (int i = 0; i < nums.size(); i++) {
                while (!st.empty() && nums[i] > nums[st.top()]) {
                    result[st.top()] = nums[i];
                    st.pop();
                }
                st.push(i);
            }
            // 最后再把结果集即result数组resize到原数组大小
            result.resize(nums.size() / 2);
            return result;
        }
    };

    这种写法确实比较直观,但做了很多无用操作,例如修改了nums数组,而且最后还要把result数组resize回去。resize倒是不费时间,是O(1)的操作,但扩充nums数组相当于多了一个O(n)的操作。

    其实也可以不扩充nums,而是在遍历的过程中模拟走了两边nums。

    // 版本二
    class Solution {
    public:
        vector<int> nextGreaterElements(vector<int>& nums) {
            vector<int> result(nums.size(), -1);
            if (nums.size() == 0) return result;
            stack<int> st;
            for (int i = 0; i < nums.size() * 2; i++) { 
                // 模拟遍历两边nums,注意一下都是用i % nums.size()来操作
                while (!st.empty() && nums[i % nums.size()] > nums[st.top()]) {
                    result[st.top()] = nums[i % nums.size()];
                    st.pop();
                }
                st.push(i % nums.size());
            }
            return result;
        }
    };

    链接:https://leetcode-cn.com/problems/next-greater-element-ii/solution/503-xia-yi-ge-geng-da-yuan-su-iidan-diao-hh9v/

    496. 下一个更大元素 I

  • 相关阅读:
    Individual Method Selection Survey rubric
    Xcode 出现Thread 1: signal SIGABRT
    C/C++生成随机数
    《深入浅出深度学习:原理剖析与python实践》第八章前馈神经网络(笔记)
    操作系统--精髓与设计原理(第八版)第三章复习题答案
    操作系统--精髓与设计原理(第八版)第二章复习题答案
    Python知识点整理
    C++ <queue>用法
    C语言结构体用法
    Mac安装Qt出现错误Could not resolve SDK Path for 'macosx'
  • 原文地址:https://www.cnblogs.com/USTC-ZCC/p/14627448.html
Copyright © 2011-2022 走看看