给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
解答一:
排序+双指针
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> ans;
int i = 0;
int j = 0;
sort(nums1.begin(), nums1.end());
sort(nums2.begin(), nums2.end());
while(i < nums1.size()&&j < nums2.size()){
if(nums1[i] > nums2[j]){
j++;
}else if(nums1[i] < nums2[j]){
i++;
}else{
ans.push_back(nums1[i]);
i++;
j++;
}
}
return ans;
}
};
解答二:
哈希表记录短数组中每个元素的个数,然后遍历较长数组,找到相同元素就将相应个数减一并推入容器中,个数为0的时候擦除该元素。
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
if(nums1.size() > nums2.size()){
return intersect(nums2, nums1);
}
vector<int> ans;
map<int, int> direction;
for(int const& i : nums1){
++direction[i];
}
for(int const& i : nums2){
if(direction.count(i)){
--direction[i];
ans.push_back(i);
}
if(direction[i] == 0){
direction.erase(i);
}
}
return ans;
}
};