问题描述:
给定两个数组,arr1
和arr2
,arr2
中的元素各不相同,
且arr2
中的所有元素都出现在arr1
中。请对arr1
中的元素进行排序,
使arr1
中元素的相对顺序和arr2
中元素的相对顺序相同,
:根绝数组2排序数组1
未在arr2
中出现的元素需要按照升序放在arr1
的末尾,最终输出排序后的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]
>
>
>
解题思路:计数排序+分组
利用字典对order2中的对应的列表元素进行计数或存储记录
剩余元素单独利用sorted进行排序
from collections import defaultdict
def function(order1,order2):dict2 =defaultdict()result = list()for i in order2:dict2[str(i)] = list()print(dict2)res = list()for num in order1:if num in order2:dict2[str(num)].append(num)else:res.append(num)print(dict2)#for item in dict2.items():#print(item.key)result = [result + v for k,vin dict2.items() ]return result[0] + sorted(res)arr1 = [2, 3, 1, 3, 2, 4, 6, 7, 9, 2, 19]
arr2 = [2, 1, 4, 3, 9, 6]
r = function(arr1,arr2)
print(r)