leetcode496. 下一个更大元素 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。

 

思路1

利用哈希表，两次遍历。

时间复杂度O(n2)。

 

代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int size1 = nums1.size(), size2 = nums2.size();
        if (size1 < 1 || size2 < 1) return res;
        // 建立哈希表
        unordered_map<int, int> myMap;
        for (int i = 0; i < size2; i++){
            myMap[nums2[i]] = i;
        }
        
        // 遍历
        for (int i = 0; i < size1; i++){
            auto index = myMap.find(nums1[i])->second;
            int j = index+1;
            for (; j < size2; j++){
                if (nums2[j] > nums1[i]) {
                    res.emplace_back(nums2[j]);
                    break;
                }
            }
            if (j == size2) res.push_back(-1);
        }
        return res;
        
    }
};

 

思路2

遍历num2，建立一个单调递增栈（栈顶到栈底是递增的)，如果进来一个元素比栈顶要大，则栈顶元素和这个新元素构成一个pair放进哈希表。然后将栈顶弹出，继续比较新元素与栈顶的关系，循环。

当num2遍历结束后，遍历num1，去哈希表找num1的元素，如果存在则将对应的数放进结果数组，如果没有则放进-1。

 

代码

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        int size1 = nums1.size(), size2 = nums2.size();
        if (size1 < 1 || size2 < 1) return res;
        // 建立哈希表
        unordered_map<int, int> myMap;
        stack<int> myStack;
        for (int i = 0; i < size2; i++){
            if (myStack.empty() || nums2[i] < myStack.top()) myStack.push(nums2[i]);
            else{
                myMap[myStack.top()] = nums2[i];
                myStack.pop();
                i--;
            }                           
        }
        for (int i = 0; i < size1; i++){
            auto it = myMap.find(nums1[i]);
            if (it == myMap.end()) res.emplace_back(-1);
            else res.emplace_back(it->second);
        }
        return res;
    }
};