数组的相对排序

/*
 * 思路：
 * 1. 使用map存储arr2中的元素及其顺序。
 * 2. 使用Stream对arr1进行排序，分两部分处理：
 *    - 在arr2中的部分按照map中的顺序排序。
 *    - 不在arr2中的部分按照自然顺序排序。
 * 3. 使用Stream.concat将两个流合并。
 */
import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.stream.Collectors;
import java.util.stream.Stream;

public class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        // 创建map存储arr2中的元素及其顺序
        Map<Integer, Integer> orderMap = new HashMap<>();
        for (int i = 0; i < arr2.length; i++) {
            orderMap.put(arr2[i], i);
        }
        // 使用Stream排序
        return Stream.concat(
            Arrays.stream(arr1).filter(orderMap::containsKey)
                .sorted((a, b) -> orderMap.get(a) - orderMap.get(b)),
            Arrays.stream(arr1).filter(a -> !orderMap.containsKey(a)).sorted()
        ).toArray();
    }
}