import java.util.*;

import java.util.stream.Collectors;

/**

* Description: 数组的相对位置

* 使用自定义比较函数实现

* Date: 2020-12-27

* Time: 10:54 AM

*/

public class RelativeSortArray1 {

Map<Integer,Integer> record;

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

record = new HashMap<>(arr2.length);

for (int i = 0; i < arr2.length; i++) {

record.put(arr2[i], i);

}

List<Integer> res = Arrays.stream(arr1).boxed().collect(Collectors.toList());

Collections.sort(res, (num1, num2) -> {

if (record.containsKey(num1) || record.containsKey(num2)) {

return record.getOrDefault(num1, 1000) - record.getOrDefault(num2, 1000);

} else {

return num1 - num2;

}

});

return res.stream().mapToInt(i -> i.intValue()).toArray();

}

public static void main(String[] args) {

int[] arr1 = new int[]{2, 3, 1, 3, 2, 4, 6, 7, 9, 2, 19};

int[] arr2 = new int[]{2, 1, 4, 3, 9, 6};

System.out.println(Arrays.toString(new RelativeSortArray1().relativeSortArray(arr1, arr2)));

}

}