本文共 1224 字，大约阅读时间需要 4 分钟。

数组的相对排序

给你两个数组 arr1 和 arr2，要求对 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]

解题思路

代码实现

class Solution:
    def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
        count = max(arr1)
        cnt = [0] * (count + 1)
        res = []
        
        for num in arr1:
            cnt[num] += 1
        
        for num in arr2:
            res.extend([num] * cnt[num])
            cnt[num] = 0
        
        for i in range(1, count + 1):
            if cnt[i]:
                res.extend([i] * cnt[i])
        
        return res

代码解释

通过这种方法，可以确保排序后的 arr1 中元素的相对顺序与 arr2 中的顺序一致，同时未在 arr2 中出现过的元素按升序排列放在末尾。