【LeetCode-数组】两个数组的交集

题目描述

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

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

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

说明：

• 输出结果中的每个元素一定是唯一的。
• 我们可以不考虑输出结果的顺序。

思路1

使用哈希表。使用哈希表 hash 记录 nums1 中的元素出现情况，无论 nums1[i] 在 nums1 中出现多少次，都令 hash[nums1[i]] 为 1。然后遍历 nums2，如果当前元素 nums2[i] 出现在 hash 中并且 hash[nums2[i]]==1，则将 nums2[i] 放进答案中，并将 hash[nums2[i]]++，这样做是为了去重。代码如下：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_map<int, int> hash;
        for(int i=0; i<nums1.size(); i++){
            if(hash[nums1[i]]<1) hash[nums1[i]]++;
        }

        vector<int> ans;
        for(int i=0; i<nums2.size(); i++){
            if(hash.count(nums2[i])!=0 && hash[nums2[i]]==1){
                ans.push_back(nums2[i]);
                hash[nums2[i]]++;
            }
        }
        return ans;
    }
};

• 时间复杂度：O(n1+n2)
• 空间复杂度：O(n1)

思路2

首先将两个数组排序，然后使用两个指针 p1, p2 分别指向两个数组，两个指针初始化为 0。

• 如果 p1<nums1.size() && p2<nums2.size(), 循环：
  • 如果 nums1[p1]==nums2[p2]，将 nums1[p1] 放进答案中（如果 nums1[p1] 没有在答案中出现过），p1++，p2++；
  • 否则，如果 nums1[p1]<nums2[p2]，则 p1++;
  • 否则，如果 nums1[p1]>nums2[p2]，则 p2++；

代码如下：

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        sort(nums1.begin(), nums1.end());
        sort(nums2.begin(), nums2.end());
        int p1 = 0, p2 = 0;
        vector<int> ans;
        unordered_set<int> hasSeen; // 记录ans中已经出现的元素，用来去重
        while(p1<nums1.size() && p2<nums2.size()){
            if(nums1[p1]==nums2[p2] ){
                if(hasSeen.count(nums1[p1])==0){
                    ans.push_back(nums1[p1]);
                    hasSeen.insert(nums1[p1]);
                }
                p1++;
                p2++;
            }else if(nums1[p1]<nums2[p2]){
                p1++;
            }else if(nums1[p1]>nums2[p2]){
                p2++;
            }
        }
        return ans;
    }
};

• 时间复杂度：O(max(n1, n2))
• 空间复杂度：O(max(n1, n2))