两个数组的交集(C++)

71 阅读1分钟

两个数组的交集

给定两个数组 nums1nums2,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序

示例 1:

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

示例 2:

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的

提示:

  • 1 <= nums1.length, nums2.length <= 1000
  • 1000```
    

我的代码:

class Solution {
public:
// 本题也就考查的一个set集合的去重特性
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> st1, st2;
        for (auto num : nums1) st1.insert(num);
        for (auto num : nums2) st2.insert(num);
        return fc(st1, st2);
    }
    vector<int> fc(unordered_set<int> st1, unordered_set<int> st2)
    {
        // 这一步的话可以节约内存
        if (st1.size() > st2.size()) return fc(st2, st1);
        vector<int> res;
        // 就这里求交集
        for (auto num:st1)
        {
            if (st2.count(num)) res.push_back(num);
        }
        return res;
    }
};

对应我的csdn文章:(2条消息) 两个数组的交集(C++)_极客李华的博客-CSDN博客