给你两个数组,arr1 和 arr2,
arr2 中的元素各不相同
arr2 中的每个元素都出现在 arr1 中
对 arr1 中的元素进行排序,使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。
题解:自定义排序,其实arr2就可以看作是这道题的排序规则,arr2中每个数字所对应的下标就是这个数的大小,当我们对arr1数组进行排序时,对于arr1中的每个元素,我们都需要在arr2中找到这个数字的下标,然后依照这个下标的大小进行排序,如果这个数字不存在于arr2中,则依照这个数的真实大小进行排序,同时需要注意的是,不存在于arr2中的数字一定大于存在于arr2中的数字。同时为了简化算法,我们可以使用哈希映射,将(arr2[ i ],i)这一键值对存放在哈希表中。
class Solution {
Map<Integer,Integer>map=new HashMap<>();
public int[] relativeSortArray(int[] arr1, int[] arr2) {
for(int i=0;i< arr2.length;i++){
map.put(arr2[i],i);
}
quickSort(arr1,0,arr1.length-1);
return arr1;
}
//快排思想
public void quickSort(int[]arr1,int start,int end){
if(start>=end)
return;
int left=start;
int right=end;
while(left<right){
while(compare(arr1[right],arr1[start])>0&&left<right){
right--;
}
while(compare(arr1[left],arr1[start])<=0&&left<right){
left++;
}
change(left,right,arr1);
}
change(left,start,arr1);
quickSort(arr1,start,left-1);
quickSort(arr1,left+1,end);
}
//自定义比较函数
public int compare(int x,int y){
if(map.containsKey(x)&&map.containsKey(y)){
return map.get(x)-map.get(y);
}else if(map.containsKey(x)){
return -1;
}else if(map.containsKey(y)){
return 1;
}else{
return x-y;
}
}
public void change(int x,int y,int[]arr1){
int temp=arr1[x];
arr1[x]=arr1[y];
arr1[y]=temp;
}
}