题目地址:
https://leetcode.com/problems/relative-sort-array/
给定两个数组 A A A和 B B B,要求将 A A A排序,使得 A A A中的数字保持其在 B B B中的相对顺序;如果不在 B B B中,则按照整数的自然顺序排序。
可以先将每个数字出现在 B B B中的下标用哈希表记下来,然后将 A A A做排序即可。代码如下:
import java.util.*;
public class Solution {
public int[] relativeSortArray(int[] arr1, int[] arr2) {
if (arr1 == null || arr1.length == 0 || arr2 == null || arr2.length == 0) {
return arr1;
}
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < arr2.length; i++) {
map.put(arr2[i], i);
}
List<Integer> arr = new ArrayList<>(arr1.length);
for (int n : arr1) {
arr.add(n);
}
// 按照以下规则排序:如果map不含两者,则直接比较自己的值;
// 若只含一个,则把那个不含的排在后面;如果都含,则比较value的值
arr.sort((i1, i2) -> {
Integer p1 = map.get(i1), p2 = map.get(i2);
if (p1 == null && p2 == null) {
return Integer.compare(i1, i2);
} else if (p1 == null) {
return 1;
} else if (p2 == null) {
return -1;
} else {
return Integer.compare(p1, p2);
}
});
for (int i = 0; i < arr1.length; i++) {
arr1[i] = arr.get(i);
}
return arr1;
}
}
时间复杂度 O ( n log n ) O(n\log n) O(nlogn),空间 O ( n ) O(n) O(n)。