双指针 | Easy | LeetCode 350. 两个数组的交集 II

链接: https://leetcode-cn.com/problems/intersection-of-two-arrays-ii
给定两个数组,编写一个函数来计算它们的交集。

说明
输出结果中每个元素出现的次数,应与元素在两个数组中出现的次数一致。
我们可以不考虑输出结果的顺序。

示例 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 的元素存储在磁盘上,磁盘内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

思路一: 排序
先对两个数组分别进行排序, 设两个指针p1和p2, 初始时分别指向第一个元素, 然后比较两个指针指向元素的值, 若*p1小于*p2, 则++p1, *p2小于*p1, 则++p2, 若两个相等, 则保存此值, 然后p1,p2分别自增。若其中一个数组遍历完成, 则程序结束。

vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
	vector<int> result;
	std::sort(nums1.begin(),nums1.end());
	std::sort(nums2.begin(),nums2.end());
	vector<int>::iterator it1 = nums1.begin();
	vector<int>::iterator it2 = nums2.begin();
	while(it1<nums1.end() && it2<nums2.end()) {
		if((*it1) < (*it2)) {
			++it1;
		} else if ((*it1) > (*it2)) {
			++it2;
		} else {
			result.push_back(*it1);
			++it1, ++it2;
		}
	}
	return result;
}

思路二: Hash
我们先在 HashMap 记录一个数组中的存在的数字和对应出现的次数。然后,我们遍历第二个数组,检查数字在 HashMap 中是否存在,如果存在且计数为正,则将该数字添加到答案并减少 HashMap 中的计数。

vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    if (nums1.size() > nums2.size()) {
        return intersect(nums2, nums1);
    }
    unordered_map<int, int> m;
    for (auto n : nums1) {
        ++m[n];
    }
    int k = 0;
    for (auto n : nums2) {
        auto it = m.find(n);
        if (it != end(m) && --it->second >= 0) {
            nums1[k++] = n;
        }
    }
    return vector(begin(nums1), begin(nums1) + k);
}

思路三:
这类似于思路1。我们不使用两个指针进行迭代,而是使用内置函数来查找公共元素。在 C++ 中,我们可以使用 set_intersection 来排序数组(或 multisets)。
在 Java 中的 retainAll 方法并不关心一个元素在另一个集合中出现的次数。

vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    sort(begin(nums1), end(nums1));
    sort(begin(nums2), end(nums2));
    nums1.erase(set_intersection(begin(nums1), end(nums1), 
        begin(nums2), end(nums2), begin(nums1)), end(nums1));
    return nums1;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值