349. 两个数组的交集
题意:
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解题思路:
具体步骤:
- 先用unordered_set存第一个数组
- 再用set存第二个(可以让数字不重复)
- 找到数字然后压入数组即可
利用set容器实现没能重复的特点,可以减去删除重复元素的操作,用这个思路我做了两个程序.
代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> set_res;
unordered_set<int> hash(nums1.begin(),nums1.end());
for (const int c:nums2)
{
if (hash.find(c) != hash.end())
set_res.insert(c);
}
vector<int> res(set_res.begin(),set_res.end());
return res;
}
};
运行过程:
代码:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
unordered_set<int> hash(nums1.begin(),nums1.end());
set<int> f(nums2.begin(), nums2.end());
for (const int c:f)
{
if (hash.find(c) != hash.end())
res.push_back(c);
}
return res;
}
};
运行结果:
总结:
两个代码思路是一样的,所以运行结果差不多.这道题要注意的是它没有说数据多大,所以不能用数组来存储,另外一个就是要利用set容器可以让重复的元素去除的特性.