给定两个数组,编写一个函数来计算它们的交集。
示例 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容器可以让重复的元素去除的特性.