介绍:给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2]
输出: [2]
示例 2:
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:
输出结果中的每个元素一定是唯一的。
我们可以不考虑输出结果的顺序。
源代码:
#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
#include <sstream>
#include<set>
using namespace std;
vector<int> intersection(vector<int>& nums1, vector<int>& nums2)
{
set<int> record(nums1.begin(), nums1.end());//这里是用nums1来初始化set类型变量,输出的是一个排序好,且无重复数据的set类型数据。
set<int>resultset;
set<int>::iterator it;
for (it = record.begin(); it != record.end(); it++)
{
cout << *it << endl;
}
for (int i = 0; i < nums2.size(); i++)
{
if (record.find(nums2[i]) != record.end())
{
//record.find()返回的是位置,当然只有查找到才能返回具体位置。
resultset.insert(nums2[i]);
}
}
return vector<int>(resultset.begin(), resultset.end());//使用set来初始化vector<int>数据
}
int main()
{
vector<int> nums1 = { 1, 3, 2, 1 }, nums2 = { 2, 2, 3 },result;
result = intersection(nums1, nums2);
cout << "最终输出:" << endl;
for (int i = 0; i < result.size(); i++)
{
cout << result[i] << endl;
}
system("pause");
return 0;
}
输出结果:
最终输出:
2
3