给定两个数组,arr1
和 arr2
,
arr2
中的元素各不相同arr2
中的每个元素都出现在arr1
中
对 arr1
中的元素进行排序,使 arr1
中项的相对顺序和 arr2
中的相对顺序相同。未在 arr2
中出现过的元素需要按照升序放在 arr1
的末尾。
示例:
输入:arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
输出:[2,2,2,1,4,3,3,9,6,7,19]
提示:
1 <= arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
arr2
中的元素arr2[i]
各不相同arr2
中的每个元素arr2[i]
都出现在arr1
中
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public int[] relativeSortArray(int[] arr1, int[] arr2) {
List<Integer> arr1List = new ArrayList<>();
for (int v : arr1) {
arr1List.add(v);
}
arr1List.sort(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
int index1 = Integer.MAX_VALUE / 2;
int index2 = Integer.MAX_VALUE / 2;
for (int i = 0; i < arr2.length; i++) {
if (arr2[i] == o1) {
index1 = i;
break;
}
}
for (int i = 0; i < arr2.length; i++) {
if (arr2[i] == o2) {
index2 = i;
break;
}
}
if (index1 == index2) {
return o1 - o2;
} else {
return index1 - index2;
}
}
});
int[] result = new int[arr1List.size()];
int i = 0;
for (Integer v : arr1List) {
result[i++] = v;
}
return result;
}
}
//leetcode submit region end(Prohibit modification and deletion)