Leetcode-350. 两个数组的交集 II

最新推荐文章于 2022-03-18 03:57:05 发布

原创最新推荐文章于 2022-03-18 03:57:05 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

Leetcode 同时被 2 个专栏收录

32 篇文章

订阅专栏

双指针

3 篇文章

订阅专栏

本文介绍了一种计算两个数组交集的方法，通过先排序再使用双指针技术，有效地处理了不同大小的数组，特别适用于其中一个数组远大于另一个的情况。讨论了不同场景下的优化策略，包括数组已排序和数据存储在磁盘上的情况。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定两个数组，写一个方法来计算它们的交集。

例如:
给定 nums1 = [1, 2, 2, 1], nums2 = [2, 2], 返回 [2, 2].

注意：

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

跟进:

如果给定的数组已经排好序呢？你将如何优化你的算法？
如果 nums1 的大小比 nums2 小很多，哪种方法更优？
如果nums2的元素存储在磁盘上，内存是有限的，你不能一次加载所有的元素到内存中，你该怎么办？

解答：

和“两个数组的交集 I”解法相同，先排序，再用双指针。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        int ptr1 = 0;
        int ptr2 = 0;
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int flag;
        int cnt = 0;
        flag = nums1.size() >= nums2.size() ? 1 : 2;
        vector<int> ret;
        if (flag == 1) {
            for (int i = 0; i<nums2.size(); i++) {
                while (ptr1<nums1.size()-1&&nums1[ptr1]<nums2[i]) ptr1++;
                if (nums1[ptr1] == nums2[i]) {
                    if (1) {
                        ret.push_back(nums2[i]);
                        cnt++;
                        ptr1++;
                    }
                }
            }
        }
        if (flag == 2) {
            for (int i = 0; i<nums1.size(); i++) {
                while (ptr2<nums2.size()-1&&nums2[ptr2]<nums1[i]) ptr2++;
                if (nums2[ptr2] == nums1[i]) {
                    if (1) {
                        ret.push_back(nums1[i]);
                        cnt++;
                        ptr2++;
                    }
                }
            }
        }
        return ret;
    }
};