【Leetcode】1122. Relative Sort Array

题目地址：

https://leetcode.com/problems/relative-sort-array/

给定两个数组$A$和$B$，要求将$A$排序，使得$A$中的数字保持其在$B$中的相对顺序；如果不在$B$中，则按照整数的自然顺序排序。

可以先将每个数字出现在$B$中的下标用哈希表记下来，然后将$A$做排序即可。代码如下：

import java.util.*;

public class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        if (arr1 == null || arr1.length == 0 || arr2 == null || arr2.length == 0) {
            return arr1;
        }
        
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < arr2.length; i++) {
            map.put(arr2[i], i);
        }
    
        List<Integer> arr = new ArrayList<>(arr1.length);
        for (int n : arr1) {
            arr.add(n);
        }
        
        // 按照以下规则排序：如果map不含两者，则直接比较自己的值；
        // 若只含一个，则把那个不含的排在后面；如果都含，则比较value的值
        arr.sort((i1, i2) -> {
            Integer p1 = map.get(i1), p2 = map.get(i2);
            if (p1 == null && p2 == null) {
                return Integer.compare(i1, i2);
            } else if (p1 == null) {
                return 1;
            } else if (p2 == null) {
                return -1;
            } else {
                return Integer.compare(p1, p2);
            }
        });
    
        for (int i = 0; i < arr1.length; i++) {
            arr1[i] = arr.get(i);
        }
        
        return arr1;
    }
}

时间复杂度$O(n\log n)$，空间$O(n)$。