LeetCode 349: 两个数组的交集

LeetCode 349: 两个数组的交集

解答

简单，关键有几个地方需要学习和记录一下

• 传入参数是容器，要求第一个参数是小容器，第二个参数是大容器
  注意getInterserction函数中，要求set1是小集合，set2是大集合，如何对传入的set1和set2进行判断和交换(如果不符合)呢？很巧的方法是重新传参

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> set1, set2;
        for (auto& num : nums1) {
            set1.insert(num);
        }
        for (auto& num : nums2) {
            set2.insert(num);
        }
        return getIntersection(set1, set2);
    }

    vector<int> getIntersection(unordered_set<int>& set1, unordered_set<int>& set2) {
    	// 注意这里
        if (set1.size() > set2.size()) {
            return getIntersection(set2, set1);
        }
        vector<int> intersection;
        for (auto& num : set1) {
            if (set2.count(num)) {
                intersection.push_back(num);
            }
        }
        return intersection;
    }
};

• 用vector初始化set等容器

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
    	// 注意初始化方法
        set<int> s1(nums1.begin(), nums1.end());
        set<int> s2(nums2.begin(), nums2.end());
        vector<int> ans;
        ans.reserve(min(s1.size(), s2.size()));
        set_intersection(s1.begin(), s1.end(),s2.begin(), s2.end(), inserter(ans, ans.begin()));
        return ans;
    }
};

• 交集，并集，差集

 set_union(A.begin(),A.end(),B.begin(),B.end(),inserter( C1 , C1.begin() ) );
//前四个参数依次是第一的集合的头尾，第二个集合的头尾。第五个参数的意思是将集合A、B取合集后的结果存入集合C中。容器A和B必须有序。

set_intersection 交集
set_union 并集
set_difference 差集

顺便看一下python的相关操作

交集：s.intersection(t) 或者 s & t
并集：s.union(t) 或者 s | t
（交集和并集可以有多个参数）
差集：s.difference(t) 或者 s - t