LeetCode简单题之数组的相对排序
題目
給你兩個數組,arr1 和 arr2,arr2 中的元素各不相同,arr2 中的每個元素都出現在 arr1 中。
對 arr1 中的元素進行排序,使 arr1 中項的相對順序和 arr2 中的相對順序相同。未在 arr2 中出現過的元素需要按照升序放在 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]
示例 2:
輸入:arr1 = [28,6,22,8,44,17], arr2 = [22,28,8,6]
輸出:[22,28,8,6,17,44]
提示:
1 <= arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
arr2 中的元素 arr2[i] 各不相同
arr2 中的每個元素 arr2[i] 都出現在 arr1 中
來源:力扣(LeetCode)
解題思路
??題目已經說明arr2中的數字都會在arr1中出現。新的序列需要按照arr2的元素的出現順序排序,假如arr1中的數字都是獨一無二的,我們可以將arr1映射到哈希表,然后遍歷arr2當arr1中的數字在哈希表里我們就先排那個數字,現在由于arr1中有很多數字是重復的,這就需要使用字典來一并將它們的頻率存儲起來;另外arr1中可能還有一些元素是arr2中沒有的,所以還需要求arr1和arr2的差集,然后將差集里的元素排序后添加到新列表的后面。
class Solution:def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:d=collections.Counter(arr1) #統計arr1中的元素頻率temp=[]for i in arr2:temp.extend([i]*d[i])for i in sorted(list(d.keys()-set(arr2))): #求差集并排序temp.extend([i]*d[i])return temp
總結
以上是生活随笔為你收集整理的LeetCode简单题之数组的相对排序的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: LeetCode简单题之找出井字棋的获胜
- 下一篇: LeetCode简单题之矩阵对角线元素的