Intersection of Two Arrays【合并两个set】

最新推荐文章于 2024-11-25 11:03:52 发布

原创最新推荐文章于 2024-11-25 11:03:52 发布 · 250 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

42 篇文章

订阅专栏

本文介绍了一种计算两个整数数组交集的高效算法。通过使用C++中的unordered_set数据结构，该算法能够快速找到两个数组中的公共元素，并确保结果中每个元素都是唯一的。文章提供了一个具体的代码实现示例。

PROBLEM：

Given two arrays, write a function to compute their intersection.

Example:
Given nums1 = [1, 2, 2, 1], nums2 = [2, 2], return [2].

Note:

Each element in the result must be unique.
The result can be in any order.

SOLVE：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> set1(nums1.begin(),nums1.end());
        vector<int> res;
        for(auto a:nums2)
            if(set1.erase(a)==1)    //erase方法如果删除会返回1，否则返回0
                res.push_back(a);
        return res;
    }
};