在arrays.sort中将lambda用作比较器

我正在尝试编写一个带有两个参数的函数，即int [] arr1和int [] arr2。然后，应按arr2中给定的顺序对arr1进行排序。例如，如果arr1为[0,1,2,3]，而arr2为[1,2,3,0]，则函数应返回[1,2,3,0]，将arr1按arr2中的顺序元素排序被索引。

我写了下面的代码，但是在Arrays.sort上仍然出现错误。我的使用不正确吗？

public int[] relativeSortArray(int[] arr1, int[] arr2) {

Map<Integer, Integer> elemToInd = new HashMap<Integer, Integer>();
for (int i = 0; i < arr2.length; i++) {
    elemToInd.put(arr2[i], i);
}

Arrays.sort(arr1, (int n1, int n2) -> elemToInd.get(n1) - elemToInd.get(n2));

return arr1;

}