題目

找尋兩個vector內的交集(有相同的元素)。

解題方法

就是先看nums1有什麼數字，存起來。再用nums2去檢查，如果有就放進去。

程式碼

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int,int> tmp;
        vector<int> ans;
        for(int i=0;i<nums1.size();i++)
        {
            if(!(tmp[nums1[i]]))
                tmp[nums1[i]] = 1;
        }
        for(int i=0;i<nums2.size();i++)
        {
            if(tmp[nums2[i]])
            {
                ans.push_back(nums2[i]);
                tmp[nums2[i]] = 0;
            }
        }
        return ans;
    }
};