说明: 输出结果中的每个元素一定是唯一的。 我们可以不考虑输出结果的顺序。
思路:哈希表
代码:
#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