给你两个数组,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
中
思路:
先遍历arr1,
找到arr1里没有在arr2中出现过的元素,把他们丢进left这个数组,
而对于在arr2里出现的元素,用dic这个哈希表记录出现的次数,key是元素的值,val是出现的次数。
然后根据dic里记录的频率,生成结果数组res,
最后把left排序,再拼接到结果数组的后面即可。
class Solution(object):
def relativeSortArray(self, arr1, arr2):
"""
:type arr1: List[int]
:type arr2: List[int]
:rtype: List[int]
"""
from collections import defaultdict
res = []
left = [] #存放没有在arr2里出现过的数字
set2 = set(arr2) #把arr2转成set,让查找的时间复杂度降低到O(1)
dic = defaultdict(int) #记录arr1中数字出现的频率
for num in arr1:
if num in set2:
dic[num] += 1
else:
left.append(num)
for num in arr2:
res += [num] * dic[num]
left.sort() #按照题目要求把没出现过的元素排序
return res + left