力扣-349. 两个数组的交集

1.题目

题目地址(349. 两个数组的交集 - 力扣（LeetCode）)

https://leetcode.cn/problems/intersection-of-two-arrays/

题目描述

给定两个数组 nums1 和 nums2 ，返回 它们的

交集

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

 

示例 1：

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

示例 2：

输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]
解释：[4,9] 也是可通过的

 

提示：

• 1 <= nums1.length, nums2.length <= 1000
• 0 <= nums1[i], nums2[i] <= 1000

2.题解

2.1 哈希集合

思路

使用unordered_set进行去重,然后查看交集即可

代码

• 语言支持：C++

C++ Code:

class Solution {
public:
    vector<int> intersection(vector<int>& nums1, vector<int>& nums2) {
        unordered_set<int> st1, st2;
        vector<int> ans;
        for(int &num : nums1){
            st1.emplace(num);
        }
        for(int &num : nums2){
            st2.emplace(num);
        }

        for(const int &num : st1){
            if(st2.count(num)){
                ans.emplace_back(num);
            }
        }
        return ans;
    }
};

复杂度分析

• 时间复杂度：\(O(m+n)\),其中\(m\)和\(n\)分别是两个数组的长度。
  使用两个集合分别存储两个数组中的元素需要\(O(m+n)\)的时间，
  遍历较小的集合并判断元素是否在另一个集合中需要\(O(\min(m,n))\)的时间，
  因此总时间复杂度是\(O(m+n)\) 。
• 空间复杂度：\(O(m+n)\),
  其中\(m\)和\(n\)分别是两个数组的长度。
  空间复杂度主要取决于两个集合。

2.2 排序+双指针

思路

代码

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;
    }
};

复杂度分析

• 时间复杂度：\(O(m\log m+n\log n)\),其中\(m\)和\(n\)分别是两个数组的长度。
  对两个数组排序的时间复杂度分别是\(O(m\log m)\)和\(O(n\log n)\) ,
  双指针寻找交集元素的时间复杂度是\(O(m+n)\) ,
  因此总时间复杂度是\(O(m\log m+n\log n)\)。
• 空间复杂度：\(O(\log m+\log n)\),其中\(m\)和\(n\)分别是两个数组的长度。
  空间复杂度主要取决于排序使用的额外空间。