开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第3天，点击查看活动详情

一 描述

1122. 数组的相对排序 - 力扣（LeetCode） (leetcode-cn.com)

给你两个数组，arr1 和 arr2，arr2 中的元素各不相同，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，数每出现一次就在字典中值+1，这样值>1，就说明出现了两次以上，需要在arr2中添加了。

添加的时候需要找到arr2中这个数字的位置，然后插入，这里应该挺耗时间。

对了，如果arr1中的数arr2中没有，单独建列表好了，最后排序后加到arr2中就好。(没审题还错了一次)

三 答案

class Solution:
    def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
        times = dict(zip(arr2,[0 for _ in range(len(arr2))]))
        new = []
        for i in arr1:
            if i not in arr2:
                new.append(i)
            else:
                times[i] += 1
                if times[i]>1:
                    here = arr2.index(i)
                    arr2.insert(here,i)
        return arr2+sorted(new) 

四 总结

首先最简单容易想到的就是自定义排序，一般的sdk都提供了自定义比较器，利用好sdk就能很容易解出此题。

但其实本题的范围不是很大，所以我们可以使用不基于比较的排序，比如计数排序。

计数排序也是排序中需要掌握的技巧之一了，如果不熟悉的话看看官解吧。

参考

用时击败99.09% 两种方法 计数器 / api - 数组的相对排序 - 力扣（LeetCode） (leetcode-cn.com)