leetcode 349.两个数组的交集

leetcode 349.两个数组的交集

题目描述

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

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

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

说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序

解题思路

使用STL中unordered_set的方法对nums1和nums2去重，然后查找两个set的交集

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty()){
            return res;
        }
        unordered_set<int> nums1Set;
        unordered_set<int> nums2Set;
        // nums1去重
        for(int i=0; i<nums1.size(); i++){
            if(nums1Set.find(nums1[i]) == nums1Set.end()){
                nums1Set.insert(nums1[i]);
            }
        }
        // num2去重
        for(int i=0; i<nums2.size(); i++){
            if(nums2Set.find(nums2[i]) == nums2Set.end()){
                nums2Set.insert(nums2[i]);
            }
        }
        // 查找交集
        for(auto num:nums1Set){
            if(nums2Set.find(num) != nums2Set.end()){
                res.push_back(num);
            }
        }

        return res;
    }
};

简化一下代码

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty()){
            return res;
        }
        // 把nums1中的元素去重
        unordered_set<int> numsSet;
        for(int i=0; i<nums1.size(); i++){
            if(numsSet.find(nums1[i]) == numsSet.end()){
                numsSet.insert(nums1[i]);
            }
        }
       // 在numsSet中查找，找到以后，先添加到目标vector中，再从集合中把该元素删除
       for(int i=0; i<nums2.size(); i++){
           if(numsSet.find(nums2[i]) != numsSet.end()){
               res.push_back(nums2[i]);
               numsSet.erase(nums2[i]);   // 删除元素
           }
       }

        return res;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步