简单题用了两个思路,成功实现:
法一:哈希表,首先哈希表中存入的是num1[i]元素出现的个数,然后再遍历nums2数组,若nums2中的元素在哈希表中也有,则将该数存入答案并将哈希表中存入的num1[i]元素的个数减一。代码如下:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
map<int,int> M;
vector<int> ans;
for(int i = 0;i < nums1.size();i++)
{
M[nums1[i]]++;
}
for(int i = 0;i < nums2.size();i++)
{
if(M[nums2[i]] > 0){
ans.push_back(nums2[i]);
M[nums2[i]]--;
}
}
sort(ans.begin(),ans.end());
return ans;
}
};
法二:双指针+排序
最先想到这个思路:
首先对两个数组进行排序,然后使用两个指针遍历两个数组。
初始时,两个指针分别指向两个数组的头部。每次比较两个指针指向的两个数组中的数字,如果两个数字不相等,则将指向较小数字的指针右移一位,如果两个数字相等,将该数字添加到答案,并将两个指针都右移一位。当至少有一个指针超出数组范围时,遍历结束。
代码如下:
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
vector<int> ans;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
for(int i = 0, j = 0;i < nums1.size() && j < nums2.size();){
if(nums1[i] == nums2[j]){
ans.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i] > nums2[j])
{
j++;
}
else if(nums1[i] < nums2[j])
{
i++;
}
}
return ans;
}
};