leetcode 350.两个数组的交集II

leetcode 350.两个数组的交集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]

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

解题思路

方法一： 选择其中的一个数组建立一个哈希表，键：键值分别对应数组中的元素和出现的次数。遍历另一个数组中的每一个元素，要是找到了次数加一，直到为零。

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty()){
            return res;
        }
		// 建立nums1数组的哈希表
        unordered_map<int, int> nums1Map;
        for(int i=0; i<nums1.size(); i++){
            if(nums1Map.find(nums1[i]) == nums1Map.end()){
                nums1Map[nums1[i]] = 1;
            }
            else{
                nums1Map[nums1[i]] += 1;
            }
        }
        // nums2数组中的每一个元素在哈希表中查找，找了以后对应的键值键一，
        // 当为零的时候表示nums1没有元素和nums2对应了
        for(int i=0; i<nums2.size(); i++){
            if(nums1Map.find(nums2[i]) != nums1Map.end()){
                if (nums1Map[nums2[i]] > 0){
                    res.push_back(nums2[i]);
                    nums1Map[nums2[i]] -= 1;
                }
            }
        }
        return res;
    }
};

方法二： 先把两个数组排序，然后通过建立两个下标，不断的比较两个元素中的值。代码如下：

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        if(nums1.empty() || nums2.empty()){
            return res;
        }

        // 两个数组排序
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        
        int nums1Index = 0, nums2Index = 0;
        while(nums1Index < nums1.size() && nums2Index < nums2.size()){
            // 判断排序后的两个数组中下标元素是否相等，相等的就是交集。
            if(nums1[nums1Index] == nums2[nums2Index]){
                res.push_back(nums1[nums1Index]);
                nums1Index++;
                nums2Index++;
            }
            else if(nums1[nums1Index] < nums2[nums2Index]){
                nums1Index++;
            }
            else{
                nums2Index++;
            }
        }
        return res;
    }
};

欢迎大家关注我的个人公众号，同样的也是和该博客账号一样，专注分享技术问题，我们一起学习进步