2020-11-02 349.两个数组的交集

349. 两个数组的交集

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

 

示例 1：

输入：nums1 = [1,2,2,1], nums2 = [2,2]
输出：[2]

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

果然菜鸡只能做做简单题了。

思路：两个数组的交集，直接先遍历数组一得到数组一中存在的元素，然后遍历数组二，获取重复的元素，但是此时每个元素都是唯一的所以考虑用unordered_set去重。

 

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        vector<int> res;
        unordered_map<int, int> hash_nums1;
        unordered_set<int> hash_nums2;
        int n1 = nums1.size(), n2 = nums2.size();
        for(int i = 0; i < n1; i ++) {
            hash_nums1[nums1[i]] ++;
        }

        for(int i = 0; i < n2; i ++) {
            if(hash_nums1.count(nums2[i])) {
                hash_nums2.insert(nums2[i]);
            }
        }
        for(auto& item : hash_nums2) {
            res.push_back(item);
        }
        return res;
    }
};

 解法二：排序+双指针  参考官方题解

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int length1 = nums1.size(), length2 = nums2.size();
        int index1 = 0, index2 = 0;
        vector<int> intersection;
        while (index1 < length1 && index2 < length2) {
            int num1 = nums1[index1], num2 = nums2[index2];
            if (num1 == num2) {
                // 保证加入元素的唯一性
                if (!intersection.size() || num1 != intersection.back()) {
                    intersection.push_back(num1);
                }
                index1++;
                index2++;
            } else if (num1 < num2) {
                index1++;
            } else {
                index2++;
            }
        }
        return intersection;
    }
};