【LeetCode】1122. Relative Sort Array 数组的相对排序（Easy）（JAVA）

【LeetCode】1122. Relative Sort Array 数组的相对排序（Easy）（JAVA）

题目地址: https://leetcode.com/problems/relative-sort-array/

题目描述：

Given two arrays arr1 and arr2, the elements of arr2 are distinct, and all elements in arr2 are also in arr1.

Sort the elements of arr1 such that the relative ordering of items in arr1 are the same as in arr2.  Elements that don’t appear in arr2 should be placed at the end of arr1 in ascending order.

Example 1:

Input: arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
Output: [2,2,2,1,4,3,3,9,6,7,19]

Constraints:

• arr1.length, arr2.length <= 1000
• 0 <= arr1[i], arr2[i] <= 1000
• Each arr2[i] is distinct.
• Each arr2[i] is in arr1.

题目大意

给你两个数组，arr1 和 arr2，

• arr2 中的元素各不相同
• arr2 中的每个元素都出现在 arr1 中

对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

解题方法

1. 采用自定义数组排序的方法：在 arr2 内的排在前，不在 arr2 的排在后；在 arr2 内的按 arr2 的 index 排序；不在 arr2 内的按大小排序
2. 因为 int[] 数组无法自定义排序，只能转为 int 的包装类 Integer

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < arr2.length; i++) {
            map.put(arr2[i], i);
        }
        Integer[] temp = new Integer[arr1.length];
        for (int i = 0; i < arr1.length; i++) {
            temp[i] = arr1[i];
        }
        Arrays.sort(temp, (a, b) -> {
        		Integer aIndex = map.get(a);
                Integer bIndex = map.get(b);
                if (aIndex == null && bIndex == null) {
                    return a - b;
                } else if (aIndex == null) {
                    return 1;
                } else if (bIndex == null) {
                    return -1;
                } else {
                    return aIndex - bIndex;
                }
        	});
        for (int i = 0; i < arr1.length; i++) {
            arr1[i] = temp[i];
        }
        return arr1;
    }
}

执行用时：3 ms, 在所有 Java 提交中击败了 39.49% 的用户
内存消耗：38.7 MB, 在所有 Java 提交中击败了 33.75% 的用户

欢迎关注我的公众号，LeetCode 每日一题更新