解法一:
set insert -- O(log n)
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
set<int> temp(nums1.begin(), nums1.end()), res;
for(int a: nums2){
if(temp.find(a)!=temp.end())
res.insert(res.end(), a);
}
return vector<int>{res.begin(), res.end()};
}
};
解法二:
class Solution {
public:
vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
vector<int> res;
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
int p1 = 0;
int p2 = 0;
while(p1<nums1.size() && p2<nums2.size()){
if(nums1[p1]<nums2[p2]) p1++;
else if(nums1[p1]>nums2[p2]) p2++;
else{
while(p1+1<nums1.size() && nums1[p1]==nums1[p1+1]) p1++;
while(p2+1<nums2.size() && nums2[p2]==nums2[p2+1]) p2++;
res.push_back(nums1[p1]);
p1++;
p2++;
}
}
return res;
}
};