[路飞][LeetCode]1122_数组的相对排序

109 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第4天,点击查看活动详情

看一百遍美女,美女也不一定是你的。但你刷一百遍算法,知识就是你的了~~

谁能九层台,不用累土起!

题目地址

题目

给你两个数组,arr1 和 arr2arr2 中的元素各不相同,arr2 中的每个元素都出现在 arr1 中。

对 arr1 中的元素进行排序,使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

示例 1:

输入: 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]

示例 2:

输入: arr1 = [28,6,22,8,44,17], arr2 = [22,28,8,6]
输出: [22,28,8,6,17,44]

提示:

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

解题思路

  • 遍历arr2,找到arr1中跟arr2中相同的项
  • 按顺序依次插入的新数组中,并将arr1中这些相同项标记为undefined
  • 完成上述操作后,我们将arr1中未被标记为undefined的筛选出来,按升序排列
  • 合并上面两个步骤产出的数组并输出

解题代码

var relativeSortArray = function(arr1, arr2) {
    const res = []
    arr2.forEach(v=>{
        for(let i =0;i<arr1.length;i++){
            if(arr1[i]===v){
                arr1[i]=undefined
                res.push(v)
            }
        }
    })
    const remain = arr1.filter(v=>v!=undefined).sort((a,b)=>a-b)
    return [...res,...remain]
};

如有任何问题或建议,欢迎留言讨论!