LeetCode 算法496 下一个更大的元素

LeetCode 算法496 下一个更大的元素

题目

给你两个 没有重复元素 的数组 nums1 和 nums2 ，其中nums1 是 nums2 的子集。

请你找出 nums1 中每个元素在 nums2 中的下一个比其大的值。

nums1 中数字 x 的下一个更大元素是指 x 在 nums2 中对应位置的右边的第一个比 x 大的元素。如果不存在，对应位置输出 -1 。

解法1 暴力求解

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
         vector<int> ans(nums1.size(), -1);
         for(int i = 0; i < nums1.size(); ++i){
             int j = 0;
             while(j < nums2.size() && nums1[i] != nums2[j]){
                 j++;
             }
             while(j < nums2.size() && nums1[i] >= nums2[j]){
                 j++;
             }
             if(j == nums2.size())
                continue;
            else
                ans[i] = nums2[j];
         }
         return ans;
     }
};

解法2 单调栈

class Solution {
public:
    vector<int> nextGreaterElement(vector<int>& nums1, vector<int>& nums2) {
        int n = nums2.size();
        map<int, int> Map1;
        stack<int> s;
        int index;
        for(int i = 0; i < n; ++i){
            while(!s.empty() && nums2[i] > nums2[s.top()]){
                index = s.top();
                s.pop();
                Map1[nums2[index]] = nums2[i];
            }
            s.push(i);
        }

        while(!s.empty()){
            Map1[nums2[s.top()]] = -1;
            s.pop();
        }

        vector<int> ans;
        for(int i = 0; i < nums1.size(); ++i){
            ans.push_back(Map1[nums1[i]]);
        }
        return ans;
    }
};