[路飞]_leetcode-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]

提示：

• 1 <= arr1.length, arr2.length <= 1000
• 0 <= arr1[i], arr2[i] <= 1000
• arr2 中的元素 arr2[i] 各不相同
• arr2 中的每个元素 arr2[i] 都出现在 arr1 中

解题思路

首先我们创建额外存储空间 temp 数组记录 arr1 数组中每个数字出现的次数。
接下来我们需要遍历 arr2 数组，根据 arr2 数组中元素的顺序，将在 arr1 中同样出现的元素按其在 arr2 中出现的顺序，结合每个数字出现的次数，将其插入结果数组中。
再接下来我们遍历 temp 数组，将未处理的元素按照升序排序放入结果数组的末尾，至此，我们就完成了 arr1 的排序。

代码实现

var relativeSortArray = function(arr1, arr2) {
    // 取值范围【0-1000】，开个大数，取一个数据1005
    const cnt = new Array(1005).fill(0);
    // 记录数字出现的次数
    for(const x of arr1) cnt[x] +=1;
    let k = 0;
    for(const x of arr2) while(cnt[x]--) arr1[k++] = x;
    // 剩余的数字按照从小到大的顺序重新排序；
    for(let i = 0; i < cnt.length; i++){
        if(cnt[i] <= 0) continue;
        while(cnt[i]--) arr1[k++] = i;
    }
    return arr1;
};

至此我们就完成了leetcode-1122-数组的相对排序