Given two arrays, write a function to compute their intersection.
分析:
题目给定两个数组,找其公共交集。该题目可以用set解决。set中存储的数据集合,每个数字仅出现一次。遍历第一个数组,将数字存入set中,遍历第二个数组,若数字在集合中出现,则将该数字存入新的数组中,最后将新集合转换成题目指定集合即可。
代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> record;
for(int i = 0; i < nums1.size(); i ++)
{
record.insert(nums1[i]);
}
set<int> resultSet;
for(int i = 0; i < nums2.size(); i++)
{
if(record.find(nums2[i]) != record.end())
resultSet.insert(nums2[i]);
}
vector<int> resultVector;
for(set<int>::iterator iter = resultSet.begin(); iter != resultSet.end(); iter ++)
{
resultVector.push_back(*iter);
}
return resultVector;
}
};