349. 两个数组的交集
难度:简单
题目
给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入: nums1 = [1,2,2,1], nums2 = [2,2] 输出: [2]
示例 2:输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4] 输出: [9,4]
思路
自己想的:先去除两个数组中的重复数字,然后再双重循环找到交集。
代码
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> target;
//先排序,然后删除重复数字
//unique()函数将重复的元素放到vector的尾部
//然后返回指向第一个重复元素的迭代器
//再用erase()函数擦除从这个元素到最后元素的所有的元素
sort(nums1.begin(),nums1.end());
nums1.erase(unique(nums1.begin(),nums1.end()),nums1.end());
sort(nums2.begin(),nums2.end());
nums2.erase(unique(nums2.begin(),nums2.end()),nums2.end());
for(int i=0;i<nums1.size();i++){
for(int j=0;j<nums2.size();j++){
if(nums1[i] == nums2[j]){
target.push_back(nums1[i]);
}
}
}
return target;
}
};
结果
执行用时 :20 ms, 在所有 C++ 提交中击败了13.40%的用户
内存消耗 :10.4 MB, 在所有 C++提交中击败了7.14%的用户