给定两个数组,编写一个函数来计算它们的交集。
1. 调用 set_intersection
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> a(nums1.begin(),nums1.end());
set<int> b(nums2.begin(),nums2.end());
vector<int> c;
set_intersection(a.begin(),a.end(),b.begin(),b.end(),inserter(c,c.begin()));
return c;
}
2. 使用set或unorder_set
//set+erase
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
unordered_set<int> u;
vector<int> answer;
for(int i:nums1)
u.insert(i);
for(int i:nums2){
auto a = u.find(i);//使用find而非遍历
if(a!=u.end()){
answer.push_back(i);
u.erase(a);//将a全部删除
}
}
return answer;
}
//double set
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
//用unordered_set对nums1的元素去重
unordered_set<int> s1;
for(auto& e : nums1)
{
s1.insert(e);
}
//用unordered_set对nums2的元素去重
unordered_set<int> s2;
for(auto& e : nums2)
{
s2.insert(e);
}
vector<int> v;
//遍历s1,如果s1中的某个元素在s2中出现,即为二者交集
for(auto& e : s1)
{
if(s2.find(e) != s2.end())
v.push_back(e);
}
return v;
}
//set+check vector
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> num1(nums1.begin(),nums1.end());
vector<int> res;
for(int i=0;i<nums2.size();i++)
{
if(num1.find(nums2[i])!=num1.end())
{
if(find(res.begin(),res.end(),nums2[i])==res.end())
res.push_back(nums2[i]);
}
}
return res;
}
3. 使用vector
//vector去重后遍历
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int>res;
if (nums1.empty() || nums2.empty())
return res;
else {
sort(nums1.begin(), nums1.end());
nums1.erase(unique(nums1.begin(), nums1.end()), nums1.end());
for (int i = 0; i < nums1.size(); i++) {
if (count(nums2.begin(), nums2.end(), nums1[i]))
res.push_back(nums1[i]);
}
return res;
}
//double vector遍历去重
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
vector<int>temp,res;
int i=0,j=0,count=0;
while(i<nums1.size()&&j<nums2.size()){
if(nums1[i]==nums2[j]){
temp.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i]>nums2[j]) j++;
else if(nums1[i]<nums2[j]) i++;
}
if(temp.size()==0) return res;
res.push_back(temp[0]);
//去重
for(int k=1;k<temp.size();k++){
if(temp[k]!=res[count]){
res.push_back(temp[k]);
count++;
}
}
return res;
}
4. 也可以用哈希表记录,用数组或map自行实现