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的填充回原数组。

int* relativeSortArray(int* arr1, int arr1Size, int* arr2, int arr2Size, int* returnSize){
    int count[1001] = {0}; //生成大小为1001的数组用来存放每个数出现的次数
    int i = 0;
    int j = 0;
    for (i = 0; i < arr1Size; i ++) { //count的下标对应arr1的值，count的值对应arr1中值出现的次数
        count[arr1[i]] ++;
    }
    for (i = 0; i < arr2Size; i++) { //找到在arr2和arr1都出现的数字
        while(count[arr2[i]] > 0) { //如果arr2的值在count所对应的下标位置出现次数大于0  那么就说明count中的这个位置存在值
            arr1[j++] = arr2[i]; //如果存在值 那就把它加到count中 按arr2的顺序排序。
            count[arr2[i]] --; //加进去了次数-1  防止进入死循环
        }
    }
    for (i = 0; i < 1001; i++) { //找在arr1 不在arr2的  不在的值按照升序排序
        while(count[i] > 0) { //同样也是找到大于0的下标 然后一直加到count中直到次数为0
            arr1[j++] = i;
            count[i]--;
        }
    }
    *returnSize = arr1Size;
    return arr1; //返回最终答案。
}