题目:
给定两个数组,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中各数组出现次数,再根据arr2数组顺序排序。剩下的升序就更简单了。
详细见代码及注释。
复杂度:
时间:就是遍历,O(n)
空间:count计数数组空间固定,复杂度O(1)。
代码:
public int[] relativeSortArray(int[] arr1, int[] arr2) {
//两数组的元素范围都在0-1000之间,共有1001个数
//count用来记录这1001个数中出现在arr1中的次数
int count[] = new int[1001];
for(int num : arr1){
//初始值为0,出现一次加一次
count[num]++;
}
//再按照arr2的顺序排,怎么实现。
//这里的逻辑也容易懂,但是不容易想到
//以坐标i来给出现在arr2中的各个arr1中的数排序
int i = 0;
for(int num:arr2){
//arr2中的每个数,只要arr1中还有,就往里加
while(count[num]>0){
arr1[i++]=num;
count[num]--;
}
}
//加完了arr2中有的,现在加没有的
//在count中剩下的值不为零的就是没有的,升序怎么实现,那就更简单了
//遍历0-1000就行了,即实现升序,又把没加入的加上了
for(int j = 0;j<1001;j++){
while(count[j]>0){
arr1[i++]=j;
count[j]--;
}
}
return arr1;
}