題目來源

496. 下一個更大元素 I

題目詳情

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

示例 1：

輸入： nums1 = [4,1,2], nums2 = [1,3,4,2].
輸出： [-1,3,-1]
解釋： nums1 中每個值的下一個更大元素如下所述：

• 4 ，用加粗斜體標識，nums2 = [1,3,4,2]。不存在下一個更大元素，所以答案是 -1 。
• 1 ，用加粗斜體標識，nums2 = [1,3,4,2]。下一個更大元素是 3 。
• 2 ，用加粗斜體標識，nums2 = [1,3,4,2]。不存在下一個更大元素，所以答案是 -1 。

示例 2：

輸入： nums1 = [2,4], nums2 = [1,2,3,4].
輸出：

• 2 ，用加粗斜體標識，nums2 = [1,2,3,4]。下一個更大元素是 3 。
• 4 ，用加粗斜體標識，nums2 = [1,2,3,4]。不存在下一個更大元素，所以答案是 -1 。

提示：

• 1 <= nums1.length <= nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 104
• nums1和nums2中所有整數 互不相同
• nums1 中的所有整數同樣出現在 nums2 中

進階： 你可以設計一個時間複雜度為 O(nums1.length + nums2.length)

相似題目

503. 下一個更大元素 II
496. 下一個更大元素 I

題解分析

解法一：單調棧

本題其實是單調棧的模板題，只不過在基礎單調棧的基礎上增加了一些前置條件的限制，比如由一個數組變為了兩個陣列。

這裡我們可以通過維護一個單調遞減棧來求解下一個更大的元素，只有棧是單調遞減的我們才能求解它的下一個更大的元素。

對於多陣列的問題，我們可以使用一個map記錄好第二個陣列中下一個最大元素的相關資訊，然後再遍歷一次第一個陣列，並從map中取出對應的結果即可。

class Solution {
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        HashMap<Integer, Integer> map = getNextGreatEle(nums2);

        int n = nums1.length;
        int[] res = new int[n];
        for(int i=0; i<n; i++){
            res[i] = map.getOrDefault(nums1[i], -1);
        }

        return res;
    }

    private HashMap<Integer, Integer> getNextGreatEle(int[] nums){
        HashMap<Integer, Integer> map = new HashMap<>();
        int n = nums.length;
        // 維護一個遞減佇列
        LinkedList<Integer> sta = new LinkedList<>();
        for(int i=0; i<n; i++){
            int now = nums[i];
            while(!sta.isEmpty() && sta.peek() < now){
                int pre = sta.poll();
                map.put(pre, now);
            }
            sta.push(now);
        }
        while(!sta.isEmpty()){
            map.put(sta.poll(), -1);
        }
        return map;
    }
}