【力扣】两个数组的交集

给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。

我的方法：对于nums1的每个元素，去nums2中寻找，找到了就删掉，并添加到结果中。时间复杂度O(n*m)，空间复杂度O(1)，只有结果的空间。这里没有考虑erase本身的时间复杂度。

STL主要函数及其时间复杂度

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        int n1 = nums1.size();
        vector<int> res;
        for (int i = 0;i < n1;i++) {
            int n2 = nums2.size();
            for (int j = 0;j < n2;j++) {
                if (nums1[i] == nums2[j]) {
                    nums2.erase(nums2.begin() + j);
                    res.push_back(nums1[i]);
                    break;
                }
            }
        }
        return res;
    }
};

题解方法1：哈希表，将较小的数组编成哈希表，思路跟我的类似，并且适用于较大数组不能装入内存的情况。时间复杂度O(n+m)，空间复杂度O(min n,m)。

• unordered_map底层是哈希表，时间复杂度O(1)
• map底层是红黑树，会自动排序，时间复杂度O(log n)
• for (int i : nums) 这种写法等同于 python 中的 for i in nums
• m.count(key) 如果有，返回1；没有，返回0

#include<unordered_map>

class Solution {
public:
    vector<int> intersect(vector<int>& nums1, vector<int>& nums2) {
        if (nums1.size() > nums2.size()) return intersect(nums2, nums1);
        unordered_map<int, int> m;
        vector<int> res;
        for (int i : nums1) ++m[i];
        for (int i : nums2) if (m.count(i) && m[i]) --m[i], res.push_back(i);
        return res;
    }
};

题解方法2：排序 + 双指针，跟双链表的排序比较像。时间复杂度O(n log n+m log m)，空间复杂度O(1)，只有结果的空间。

C++居然给了sort()。void sort (first, last, comp)，三个参数分别是开头，结尾和用来比较的函数。函数可以不填，默认是小于运算（升序）。

sort() 只对 array、vector、deque 这 3 个容器提供支持。底层实现是快速排序，故无法保证相同元素的相对位置不发生改变，可以选用 stable_sort() 排序函数。

#include<algorithm>

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;
        int index1 = 0, index2 = 0;
        while (index1 < length1 && index2 < length2) {
            if (nums1[index1] < nums2[index2]) {
                index1++;
            } else if (nums1[index1] > nums2[index2]) {
                index2++;
            } else {
                intersection.push_back(nums1[index1]);
                index1++;
                index2++;
            }
        }
        return intersection;
    }
};

作者：LeetCode-Solution
链接：https://leetcode.cn/problems/intersection-of-two-arrays-ii/solution/liang-ge-shu-zu-de-jiao-ji-ii-by-leetcode-solution/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。