leetcode 349. 两个数组的交集 哈希表 & vector双指针去重-爱代码爱编程
349. 两个数组的交集
给定两个数组 nums1
和 nums2
,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
输入:nums1 = [1,2,2,1], nums2 = [2,2] 输出:[2]
思路1:
vector 去重 + 双指针遍历
vector 去重操作:
1 先排序:
sort(nums1.begin(),nums1.end());
2 去重:(注意unique用法)
nums1.erase(unique(nums1.begin(),nums1.end()),nums1.end());
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
//先对两个数组进行去重排序
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
nums1.erase(unique(nums1.begin(),nums1.end()),nums1.end());
nums2.erase(unique(nums2.begin(),nums2.end()),nums2.end());
vector<int> ans;
int n1=nums1.size();
int n2=nums2.size();
//用双指针遍历
for(int i=0,j=0;i<n1 && j<n2;)
{
if(nums1[i]==nums2[j])
{
ans.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i]<nums2[j])
{
i++;
}
else
{
j++;
}
}
return ans;
}
};
思路二:
哈希映射——>unordered_set
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
//先对一个元素集合进行遍历
//把元素放入哈希表中
//构造num1的哈希表
unordered_set<int> mp(nums1.begin(),nums1.end());
//结果哈希表
unordered_set<int> result_set;
//把num1和num2的相同元素放入num2中
for(auto num : nums2)
{
if(mp.find(num)!= mp.end())
{
result_set.insert(num);
}
}
return vector<int>(result_set.begin(),result_set.end());
}
};