LeetCode：下一個更大元素I【31】

題目描述

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

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

示例 1:

輸入: nums1 = [4,1,2], nums2 = [1,3,4,2].
輸出: [-1,3,-1]
解釋:
    對於num1中的數字4，你無法在第二個陣列中找到下一個更大的數字，因此輸出 -1。
    對於num1中的數字1，第二個陣列中數字1右邊的下一個較大數字是 3。
    對於num1中的數字2，第二個陣列中沒有下一個更大的數字，因此輸出 -1。

示例 2:

輸入: nums1 = [2,4], nums2 = [1,2,3,4].
輸出: [3,-1]
解釋:
    對於num1中的數字2，第二個陣列中的下一個較大數字是3。
    對於num1中的數字4，第二個陣列中沒有下一個更大的數字，因此輸出 -1。

題目分析

　　認真審題是很重要的！題目中說的是陣列1中的元素是陣列2的子集，那說明陣列2中都能找到1中的元素，然後下一個更大元素是指 x 在 nums2 中對應位置的右邊的第一個比 x 大的元素。那我們先處理nums2，把每一個元素以及右邊大於它的第一個元素存到一個數據結構Map中，然後構造結果集，在按照nums1的元素順序從Map中取出來就可以了。

　　我們還是畫圖來模擬一下這個過程，模擬過程只是方便我們去理解如何利用棧結構實現下一個的問題：

Java題解

public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        HashMap<Integer,Integer> map = new HashMap<>();
        Stack<Integer> stack = new Stack<>();
        int[] res = new int[nums1.length];

        for(int num:nums2)
        {
            while(!stack.isEmpty()&&stack.peek()<num)
            {
                map.put(stack.pop(),num);
            }
            stack.push(num);

        }

        while (!stack.isEmpty())
        {
            map.put(stack.pop(),-1);
        }

        for(int i=0;i<nums1.length;i++)
            res[i]=map.get(nums1[i]);
        return res;

    }