LeetCode1122：数组的相对排序

目录

一、题目

二、示例

三、思路

四、代码

---

一、题目

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

提示：

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

三、思路

• 第一步，利用字典nums将排序后的数组arr1的元素和出现的次数记录；
• 第二步，按照数组arr2的相对顺序，将arr1的元素存入arr中，此时利用函数newarray，append对应元素及个数；
• 第三步，判断：如果去重后的数组arr1和数组arr2的长度相等，即返回arr；否则，求二者差集，将arr2中没有arr1中有的元素append到数组arr中。

四、代码

class Solution:
    def relativeSortArray(self, arr1, arr2):
        """
        :type arr1: List[int]
        :type arr2: List[int]
        :rtype: List[int]
        """

        # 1、字典存储arr1中的数字出现的次数
        nums = {}
        arr1.sort()
        for i in arr1:
            if i not in nums:nums[i] = 1
            else:nums[i] += 1

        # 2、定义一个函数newarray
        def newarray(num, count):
            while count > 0:
                arr.append(num)
                count -= 1

        # 3、按照arr2中的数字出现的次序存入arr中，函数newarray相当于append
        arr = []
        for j in arr2:
            if j in arr1:newarray(j, nums[j])

        # 4、判断：
        # 如果去重后的arr1和arr2长度相等，即返回arr；
        # 否则，求二者差集，并将arr1中存在且arr2中未出现的存入arr中
        if len(set(arr1)) == len(set(arr2)):
            return arr
        else:
            tmp = []
            # arr1中有，arr2中没有。即两数组的差集
            for k in list(set(arr1).difference(set(arr2))):
                tmp.append(k)
            tmp.sort()
            for l in tmp:
                newarray(l, nums[l])
            return arr


if __name__ == '__main__':
    # arr1 = [2, 3, 1, 3, 2, 4, 6, 7, 9, 2, 19]
    # arr2 = [2, 1, 4, 3, 9, 6]
    arr1 = [2, 21, 43, 38, 0, 42, 33, 7, 24, 13, 12, 27, 12, 24, 5, 23, 29, 48, 30, 31]
    arr2 = [2, 42, 38, 0, 43, 21]
    s = Solution()
    ans = s.relativeSortArray(arr1, arr2)
    print(ans)