给定两个数组,编写一个函数来计算它们的交集。
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
通过次数260,391提交次数472,744
题目分析
用一个vector(数组,如果学过线性代数理解成行向量或者列向量吧~)记录答案。
首先很容易可以想到先遍历一个数组,用哈希表记录数字出现的次数。
然后再遍历后一个数组的每一个数字,若哈希表中对应出现次数不为0,那么向vector中加入这个元素,然后将其出现次数减一。
然后很容易可以想到从小到大排序后,维护两个指针i和j,分别指向nums1的左侧元素和nums2的右侧元素,然后走就完了
AC代码
class Solution {
public:
vector<int> intersect(vector<int>& nums1, vector<int>& nums2)
{
vector<int> ret;
sort(nums1.begin(),nums1.end());
sort(nums2.begin(),nums2.end());
int i = 0,j =0;
while(i<nums1.size()&&j<nums2.size())
{
if(nums1[i]==nums2[j])
{
ret.push_back(nums1[i]);
i++;
j++;
}
else if(nums1[i]>nums2[j]) j++;
else i++;
}
return ret;
}
};