領釦LintCode演算法問題答案-1206. 下一個更大的數 I

1206. 下一個更大的數 I

描述

你有兩個陣列 nums1和nums2 （互不重複） ，其中nums1是nums2的子集。 在nums2的相應位置找到nums1所有元素的下一個更大數字。

nums1中的數字x的下一個更大數字是nums2中x右邊第一個更大的數字。 如果它不存在，則為此數字輸出-1。

1. nums1和nums2中的所有數字都是唯一的。
2. nums1和nums2的長度不超過1000。

樣例 1：

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

樣例 2：

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

題解

public class Solution {
    /**
     * @param nums1: an array
     * @param nums2: an array
     * @return:  find all the next greater numbers for nums1's elements in the corresponding places of nums2
     */
    public int[] nextGreaterElement(int[] nums1, int[] nums2) {
        // Write your code here
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums2.length; i++) {
            map.put(nums2[i], i);
        }

        int[] ret = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            ret[i] = -1;
            for (int j = map.get(nums1[i]); j < nums2.length; j++) {
                if (nums2[j] > nums1[i]) {
                    ret[i] = nums2[j];
                    break;
                }
            }
        }

        return ret;
    }
}

原題連結點這裡

鳴謝

非常感謝你願意花時間閱讀本文章，本人水平有限，如果有什麼說的不對的地方，請指正。
歡迎各位留言討論，希望小夥伴們都能每天進步一點點。