问题描述:
给定两个数组,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)