349. 两个数组的交集

    xiaoxiao2022-07-01  117

    给定两个数组,编写一个函数来计算它们的交集。

    示例 1:

    输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2]

    示例 2:

    输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [9,4]

    说明:

    输出结果中的每个元素一定是唯一解法一:

    解法一:

    class Solution { public: vector<int> intersection(vector<int>& nums1, vector<int>& nums2) { vector<int> curr; unordered_map<int, int> res; for(auto c : nums1) ++res[c]; for(auto c : nums2) { if(res[c] != 0) { curr.push_back(c); res[c] = 0; } } return curr; } };

    解法二:

    class Solution { public: vector<int> intersection(vector<int>& nums1, vector<int>& nums2) { vector<int> res; set<int> st; for_each(nums1.begin(),nums1.end(),[&st](int i){st.insert(i);}); for(int e:st) if(find(nums2.cbegin(),nums2.cend(),e)!=nums2.cend()) res.push_back(e); return res; } };

     

    最新回复(0)