單調棧系列-Leetcode 496. 下一個更大元素 I

問題描述

給定兩個 沒有重複元素 的數組 nums1 和 nums2 ，其中nums1 是 nums2 的子集。找到 nums1 中每個元素在 nums2 中的下一個比其大的值。

nums1 中數字 x 的下一個更大元素是指 x 在 nums2 中對應位置的右邊的第一個比 x 大的元素。如果不存在，對應位置輸出 -1 $^{[1]}$。

解題報告

求解每個元素的下一個更大元素，採用 單調遞減棧 來存儲數組中的元素，當前元素小於棧頂元素時，壓棧；當前元素大於棧頂元素時，當前元素即爲棧頂元素的下一個更大元素，出棧，記錄棧頂元素的下一個更大元素，直到棧頂元素小於當前元素。

將每個元素的下一個更大元素存到哈希表中，之後可以在 $O(1)$ 的時間複雜度內查詢。

實現代碼

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

        for(int i=0;i<nums1.size();i++){
            if(mp.find(nums1[i])!=mp.end())
                nums1[i]=mp[nums1[i]];
            else
                nums1[i]=-1;
        }
        return nums1;
    }
};

參考資料

[1] 單調棧系列-Leetcode 496. 下一個更大元素 I