1122. 数组的相对排序

给你两个数组，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.length, arr2.length <= 1000
• 0 <= arr1[i], arr2[i] <= 1000
• arr2 中的元素 arr2[i] 各不相同
• arr2 中的每个元素 arr2[i] 都出现在 arr1 中

来源：力扣（LeetCode）

最简单直观的方法，先遍历arr2，然后用一个指针指向arr1并向后移动，凡是和当前遍历到的arr2的元素相同就交换位置。

arr2遍历结束后，arr1中的指针后面的元素全部都是arr2中没有的，直接排序即可。

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int k=0;
        for(int i=0;i<arr2.length;i++){
            int cur=arr2[i];
            for(int j=k;j<arr1.length;j++){
                if(arr1[j]==cur){
                    swap(arr1,k,j);
                    k++;
                }
            }
        }
        Arrays.sort(arr1,k,arr1.length);
        return arr1;
    }
    public void swap(int[] nums,int i,int j){
        int temp=nums[i];
        nums[i]=nums[j];
        nums[j]=temp;
    }
}

另外的方法：计数排序，思路写在代码的注释里

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int max=0;
        for(int num:arr1){
            max=Math.max(max,num);
        }
        int[] count=new int[max+1];
        //统计arr1数组中每个元素出现的次数，统计结束后arr1就不会提供我们需要的信息了，可以被覆盖
        for(int num:arr1){
            count[num]++;
        }
        //按元素在arr2出现的顺序和count数组所保存的出现次数，从头放在arr1中
        int k=0;
        for(int num:arr2){
            for(int i=0;i<count[num];i++){
                arr1[k++]=num;
            }
            //保存后记得给count归零，防止在后面计算没在arr2出现的元素时重复计算
            count[num]=0;
        }
        //计算没在arr2中出现的元素
        for(int i=0;i<=max;i++){
            if(count[i]!=0){
                for(int j=0;j<count[i];j++){
                    arr1[k++]=i;
                }
            }
        }
        return arr1;
    }
}