「这是我参与2022首次更文挑战的第22天,活动详情查看:2022首次更文挑战」。
题目描述:
1122. 数组的相对排序 - 力扣(LeetCode) (leetcode-cn.com)
给你两个数组,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]
示例二
输入: arr1 = [28,6,22,8,44,17], arr2 = [22,28,8,6]
输出: [22,28,8,6,17,44]
提示:
- 1 <= arr1.length, arr2.length <= 1000
- 0 <= arr1[i], arr2[i] <= 1000
- arr2 中的元素 arr2[i] 各不相同
- arr2 中的每个元素 arr2[i] 都出现在 arr1 中
思路分析
计数排序
注意到本题中元素的范围为 [0, 1000][0,1000],这个范围不是很大,我们也可以考虑不基于比较的排序,例如「计数排序」。
具体步骤如下:
1、定义一个数组 hash,长度为 1001,用于模拟哈希表。
2、将 arr1 中的所有元素存放在 hash 数组中,其中的键为 arr1 中的元素,值为 arr1 中的元素出现的次数;
3、定义索引 i,初始化为 0,用于重置数组 arr1 中的元素值。
4、遍历数组 arr2,只要 arr2 中的元素在数组 hash 中存在,则将其赋值给 arr1,并对该元素在 hash 中出现的频次减一;
5、针对不是数组 arr2 中的元素,遍历整个数组 hash,只要其元素出现的次数在一次及以上,将其赋值给 arr1,并将该元素在 hash 中出现的频次减一。
AC代码
class Solution {
public int[] relativeSortArray(int[] arr1, int[] arr2) {
int i = 0;
int [] hash = new int [1001];
for (int n : arr1) {
hash[n]++;
}
for (int n : arr2) {
while (hash[n] > 0) {
arr1[i++] = n;
hash[n]--;
}
}
for (int n = 0; n < hash.length; ++n) {
while (hash[n] > 0) {
arr1[i++] = n;
hash[n]--;
}
}
return arr1;
}
}