C++ 两个数组的交集

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

思路:哈希表

 代码:

#include<iostream>
#include<vector>
#include<string.h>
#include<unordered_set>//一定要引用头文件
using namespace std;
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
    {
        unordered_set<int> result_set; 
        unordered_set<int> nums_set(nums1.begin(), nums1.end());
        for (int num : nums2) //依次让num=nums[i]
        {
            
            if (nums_set.find(num) != nums_set.end())//判断nums_set有没有nums_2的元素 
            {
                result_set.insert(num);
            }
        }
        return vector<int>(result_set.begin(), result_set.end());
    }
void print(vector<int>&C)
{
    for(int i=0;i<C.size();i++)
    {
        cout<<C[i]<<' ';
    }
}
    int main()
    {
        cout<<"输入A,B容器的大小:"<<endl;
        int A_len,B_len;
        cin>>A_len;
        cin>>B_len;
        vector<int>A;
        vector<int>B ;
        cout<<"输入元素"<<endl;
        for (int i=0;i<A_len;i++)
        {           
            int a;
            cin>>a;
            A.push_back(a);
        }
        cout<<"输入元素"<<endl;
        for (int i=0;i<B_len;i++)
        {
            int b;
            cin>>b;
            B.push_back(b);
        }
        vector<int>C=intersection(A,B);
        cout<<"重复的元素为:"<<endl;
        print(C);

        return 0;
    }

运行结果:

输入A,B容器的大小:
3  
3
输入元素
1
2
3
输入元素
2
2
3
重复的元素为:
3 2


版权声明:本文为wangcheng_BI原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。