LeetCode--350. 两个数组的交集 II(C++描述)

// Source : https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
// Date : 2021-10-4

题目描述:给定两个数组,编写一个函数来计算它们的交集。
示例 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 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?

题目分析:先对元素进行排序,相比于该题的第一个版本,此版本可以将两个数组直接排序,然后从头开始遍历两个数组,它允许元素重复,故当一个数组的对应元素小于另一个数组的元素的时候可以将较小元素所在数组的index向后移动。继续比较,碰到相等的时候将元素放入最终数组,否则就向后移动对应的Index。

代码:

class Solution {
    
    
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
    
    
        //分别排序两个数组
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        //求出两个数组的长度
        int length1 = nums1.size(), length2 = nums2.size();
        vector<int> intersection;
        //定义两个下标,分别从0开始
        int index1 = 0, index2 = 0;
        //当下标都合法的时候,把较小元素的下标后移
        while (index1 < length1 && index2 < length2) 
        {
    
    
            if (nums1[index1] < nums2[index2]) 
                index1++;
            else if (nums1[index1] > nums2[index2]) 
                index2++;
            //如果二者相等,就将元素放入数组,然后将二者的index同时后移
            else 
            {
    
    
                intersection.push_back(nums1[index1]);
                index1++;
                index2++;
            }
        }
        return intersection;
    }
};

猜你喜欢

转载自blog.csdn.net/qq_44614524/article/details/120601818