数组的相对排序

/*
 * 思路：
 * 1. 创建一个映射，记录arr2中每个元素的位置。
 * 2. 使用Java Stream对arr1中的元素进行排序：
 *    - 先根据arr2中的顺序排序，
 *    - 再根据自然顺序排序。
 */
import java.util.*;
import java.util.stream.*;

public class RelativeSortArrayStream {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        Map<Integer, Integer> rank = IntStream.range(0, arr2.length)
                .boxed()
                .collect(Collectors.toMap(i -> arr2[i], i -> i));

        return Arrays.stream(arr1)
                .boxed()
                .sorted((a, b) -> {
                    if (rank.containsKey(a) && rank.containsKey(b)) {
                        return rank.get(a) - rank.get(b);
                    } else if (rank.containsKey(a)) {
                        return -1;
                    } else if (rank.containsKey(b)) {
                        return 1;
                    } else {
                        return a - b;
                    }
                })
                .mapToInt(i -> i)
                .toArray();
    }
}