Leetcode 1122. 数组的相对排序

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]

思路

• 这道题目用python解决就非常方便， 需要利用到index， 下面举个栗子：

arr1 = [1, 2, 3], arr2 = [3, 2, 1]
如果arr1.sort(key=arr2.index)
最后arr1 = [3, 2, 1]

• 对于index方法我们可以理解成按照下标索引值大小来排序， 在上面的例子中， arr2中， 3的index=0， 2的index=1， 1的index=2， 因为0 < 1 < 2， 所以这三者最终排序成3 2 1
• 简单的来理解， 就是一个数组可以按照另外一个数组的相对顺序来排序

代码 —— python

class Solution:
    def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
        arr_margin = list(set(arr1) - set(arr2))
        arr_margin.sort()
        return sorted(arr1, key=(arr2 + arr_margin).index)