题目:
计数排序法:以空间提升时间
- 借助辅助数组来计数,一是记录下数组二的数在数组一中对应出现的次数,即频度;二是记录下数组一除了数组二的数以外其他数按序排序的结果。
- 遍历数组二,从辅助数组对应下标下取出对应频度个数字加入结果数组,完成这一遍遍历后,顺序输出辅助数组的非0值对应的下标,即完成排序。
- 代码分析
小tips:新建辅助数组时,题目提示arr2.length <= 1000,但是不用建1001个数组空间,我们可以遍历数组1得出其最大值,新建(upper+1)个即可
int upper = 0;
for (int x : arr1) {
upper = Math.max(upper, x);
}
- 得出数组1个数字出现的频度
for (int x : arr1) {
++frequency[x];
}
- 以数组2的顺序,输出频度次个
for (int x : arr2) {
for (int i = 0; i < frequency[x]; ++i) {
ans[index++] = x;
}
frequency[x] = 0;
}
- 把数组2没有出现的其他数字按序输出
for (int x = 0; x <= upper; ++x) {
for (int i = 0; i < frequency[x]; ++i) {
ans[index++] = x;
}
}
整理起来:
class Solution {
public int[] relativeSortArray(int[] arr1, int[] arr2) {
int upper = 0;
for (int x : arr1) {
upper = Math.max(upper, x);
}
int[] frequency = new int[upper + 1];
for (int x : arr1) {
++frequency[x];
}
int[] ans = new int[arr1.length];
int index = 0;
for (int x : arr2) {
for (int i = 0; i < frequency[x]; ++i) {
ans[index++] = x;
}
frequency[x] = 0;
}
for (int x = 0; x <= upper; ++x) {
for (int i = 0; i < frequency[x]; ++i) {
ans[index++] = x;
}
}
return ans;
}
}