[路飞]_leetcode-1122-数组的相对排序

「这是我参与2022首次更文挑战的第11天，活动详情查看：2022首次更文挑战」

[题目地址] [B站地址]

给你两个数组，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 中

解题思路

因为本题数组中的元素都是大于等于 0 的整数，所以我们可以使用计数排序进行解题。
如果你对计数排序还不了解的话，可以看我之前的你真的懂排序吗？这篇文章，这里我们不再重复讲了。
首先我们创建额外存储空间 temp 数组记录 arr1 数组中每个数字出现的次数。
接下来我们需要遍历 arr2 数组，根据 arr2 数组中元素的顺序，将在 arr1 中同样出现的元素按其在 arr2 中出现的顺序，结合每个数字出现的次数，将其插入结果数组中。
再接下来我们遍历 temp 数组，将未处理的元素按照升序排序放入结果数组的末尾，至此，我们就完成了 arr1 的排序。

代码实现

var relativeSortArray = function(arr1, arr2) {
  // 创建额外存储空间存储每个值的出现次数
  const temp = []
  // 遍历输入数组，并计数
  for(let i = 0;i<arr1.length;i++){
    const num = arr1[i]
    if(temp[num]===undefined) temp[num] = 1
    else temp[num]++
  }
  // 创建结果数组
  const res = []
  // 按照 arr2 中元素的顺序放入 arr1 中相同的元素
  for(let i = 0;i<arr2.length;i++){
    const num = arr2[i]
    while(temp[num]){
      res.push(num)
      temp[num]--
    }
  }
  // 将剩余元素从小到大插入结果数组
  for(let i = 0;i<temp.length;i++){
    if(temp[i]===0) continue
    while(temp[i]){
      res.push(i)
      temp[i]--
    }
  }
  // 返回结果数组
  return res;
};

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

如有任何问题或建议，欢迎留言讨论！👏🏻👏🏻👏🏻