leetcode 1122. 數組的相對排序

【題目】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 中

【解題思路1】計數排序

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int[] count = new int[1001];
        for(int i : arr1){
            count[i]++;
        }
        int i = 0;
        for(int j : arr2){
            while(count[j] > 0){
                arr1[i++] = j;
                count[j]--;
            }
        }
        for(int j = 0; j <= 1000; j++){
            while(count[j] > 0){
                arr1[i++] = j;
                count[j]--;
            }
        }
        return arr1;
    }
}

【解題思路2】Arrays.sort()改寫