LeetCode 1122. 陣列的相對順序
阿新 • • 發佈:2020-11-14
題目描述
給你兩個陣列,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
中
來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/relative-sort-array
思路解析
arr1
中的數字可以分為兩部分,第一部分為在 arr2
中出現過的數字,第二部分為 arr2
中未出現的數字。
使用一個雜湊表來儲存 arr1
中每個元素出現的次數,然後遍歷 arr2
陣列,可以按 arr2
中相對順序對 arr1
中第一部分的數字完成排序。
再將第二部分的數字排序即可。
程式碼實現
class Solution { public: vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) { unordered_map<int, int> dict; vector<int> result; for(int num : arr1) dict[num]++; for(int num : arr2) { while(dict[num] > 0) { result.push_back(num); dict[num]--; } } int ordered_num = result.size(); for(auto p : dict) { while(p.second > 0) { result.push_back(p.first); p.second--; } } sort(result.begin() + ordered_num, result.end()); return result; } };