Leetcode 兩個陣列的交集 II(c++)
阿新 • • 發佈:2018-11-22
給定兩個陣列,編寫一個函式來計算它們的交集。
示例 1:
輸入: nums1 = [1,2,2,1], nums2 = [2,2]
輸出: [2,2]
示例 2:
輸入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
輸出: [4,9]
說明:
輸出結果中每個元素出現的次數,應與元素在兩個陣列中出現的次數一致。
我們可以不考慮輸出結果的順序。
採用map儲存和查詢!
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
map<int,int> MapNums;
for (int i=0;i<nums1.size();++i)
{
++ MapNums[nums1[i]];
}
vector<int> vec;
for (int j=0;j<nums2.size();++j){
if(MapNums[nums2[j]]>0){
vec. push_back(nums2[j]);
--MapNums[nums2[j]];
}
}
return vec;
}
};