剑指 Offer II 075. 数组相对排序
给定两个数组,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 中
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/0H97ZC
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
对于重复数字很多的长数组,利用计数排序往往是比其他排序方法更快的。
计数排序的思想是遍历一次数组记录下最大最小值,然后把重复数字用大小为最大值-最小值的数组哈希表记录下来,然后按照顺序把数字放进数组中。
本题要按照题目规定的顺序进行排列,因此直接将重复数字用数组哈希表记录,哈希表的长度应该为1001(题目数组数字最大值),然后将数组按照规定顺序进行排列,最后还没排列的数字再利用哈希表进行排列即可。
题解(Java)
class Solution {
public int[] relativeSortArray(int[] arr1, int[] arr2) {
int[] counts = new int[1001];
for (int i : arr1) {
counts[i]++;
}
int j = 0;
for (int i : arr2) {
while (counts[i] > 0) {
arr1[j++] = i;
counts[i]--;
}
}
for (int i = 0; i < counts.length; i++) {
while (counts[i] > 0) {
arr1[j++] = i;
counts[i]--;
}
}
return arr1;
}
}