力扣刷题记录#数组#简单#1122数组的相对排序

题目描述

给出两个数组，arr1 和 arr2，

• arr2 中的元素各不相同
• arr2 中的每个元素都出现在 arr1 中

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

注意：

• arr1.length, arr2.length <= 1000
• 0 <= arr1[i], arr2[i] <= 1000
• arr2 中的元素 arr2[i] 各不相同
• arr2 中的每个元素 arr2[i] 都出现在 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]

解答

classSolution(object):defrelativeSortArray(self,arr1,arr2):""":type arr1:List[int]:type arr2:List[int]:rtype:List[int]""" len1=len(arr1)len2=len(arr2)iflen1==len2:returnarr2 # 统计arr1中各个数出现的次数 counts1=dict()fornum in arr1:ifnum not in counts1.keys():counts1[num]=0counts1[num]+=1# 统计arr2中各个数出现的次数 counts2=dict()fornum in arr2:ifnum not in counts2.keys():counts2[num]=0counts2[num]+=1# 没在arr2中出现的数（无重复） rest=list(set(arr1)-set(arr2))# 将arr1中包含arr2的数排序，按arr2的顺序 ans=list()fornum in arr2:count=counts1[num]ans=ans+[num]*count # 将没在arr2中出现的数排序 rest_all=list()fornum in rest:count=counts1[num]rest_all=rest_all+[num]*count rest_all.sort()returnans+rest_all