两个数组的交集 II(简单)
给你两个整数数组 nums1 和 nums2 ,请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。可以不考虑输出结果的顺序。
题解
1.哈希map映射
记录第一个数组中每个元素出现的次数,遍历第二个数组,查找在哈希表中该元素出现次数是否大于0,如果大于0,说明该元素是交集,并把次数减一
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
// if(nums1.size() > nums2.size())
// return intersect(nums2,nums1);
unordered_map<int,int> s;
for(int i = 0;i <nums1.size();i++){
s[nums1[i]]++;
}
vector<int> ans;
for(int j = 0;j <nums2.size();j++){
if(s[nums2[j]] > 0){
s[nums2[j]]--;
ans.push_back(nums2[j]);
}
}
return ans;
}
};
2.双指针
排序后遍历
如果两个数组是有序的,则可以使用双指针的方法得到两个数组的交集。
首先对两个数组进行排序,然后使用两个指针遍历两个数组。
初始时,两个指针分别指向两个数组的头部。每次比较两个指针指向的两个数组中的数字,如果两个数字不相等,则将指向较小数字的指针右移一位,如果两个数字相等,将该数字添加到答案,并将两个指针都右移一位。当至少有一个指针超出数组范围时,遍历结束。
解析传送门
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int i = 0,j = 0;
vector<int> ans;
while(i < nums1.size() && j < nums2.size()){
if(nums1[i] == nums2[j]) {
ans.push_back(nums2[j]);
i++;
j++;
}
else if(nums1[i] > nums2[j]) j++;
else i++;
}
return ans;
}
};