349. Intersection of Two Arrays

最新推荐文章于 2025-08-13 21:50:19 发布

原创最新推荐文章于 2025-08-13 21:50:19 发布 · 121 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #Array #Intersection

LeetCode 专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种使用STL计算两个数组交集的算法。通过将较小数组转换为集合，遍历较大数组并检查元素是否存在于集合中，来高效地找到公共元素。最终返回不重复的交集结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

Example 1:

Input: nums1 = [1,2,2,1], nums2 = [2,2]
Output: [2]

Example 2:

Input: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
Output: [9,4]

Note:

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

使用STL，十分方便。

vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
    if(nums1.size() > nums2.size())
    {
        vector<int> temp = nums1;
        nums1 = nums2;
        nums2 = temp;
    }
    set<int> n1;
    set<int> contemporary;
    for(auto a : nums1)n1.insert(a);
    for(auto b : nums2)
        if(n1.find(b) != n1.end())contemporary.insert(b);
    vector<int> result;
    for(set<int>::iterator it = contemporary.begin(); it != contemporary.end(); it++)result.push_back(*it);
    return result;
}