350. 两个数组的交集 II 🟩

题目

LeetCode 简单

题目

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

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

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

题解

方法：Map

时间复杂度：O(m+n)

m 和 n 分别是两个数组的长度。需要遍历两个数组并对哈希表进行操作，哈希表操作的时间复杂度是 O(1)，因此总时间复杂度与两个数组的长度和呈线性关系。
空间复杂度：O(min(m,n))

m 和 n 分别是两个数组的长度。对较短的数组进行哈希表的操作，哈希表的大小不会超过较短的数组的长度。为返回值创建一个数组 intersection，其长度为较短的数组的长度。

function intersect(nums1, nums2) {
    if (nums1.length > nums2.length) {
        return intersect(nums2, nums1)
    }
    let map = new Map()
    nums1.forEach(item => {
        map.set(item, map.has(item) ? map.get(item) + 1 : 1)
    })
    let result = []
    for (let i = 0; i < nums2.length; i++) {
        if (map.has(nums2[i]) && map.get(nums2[i]) > 0) {
            result.push(nums2[i])
            map.set(nums2[i], map.get(nums2[i]) - 1)
        }
    }
    return result
}

console.log('🌰', intersect([4, 9, 5],  [9, 4, 9, 8, 4]))
console.log('🌰', intersect([1, 2, 2, 1], [2, 2]))

执行结果

🌰 [ 9, 4 ]
🌰 [ 2, 2 ]

方法：排序 + 双指针

时间复杂度：O(mlogm+nlogn)

其中 m 和 n 分别是两个数组的长度。对两个数组排序的时间复杂度分别是 O(mlogm) 和 O(nlogn)，双指针寻找交集元素的时间复杂度是 O(m+n)，因此总时间复杂度是 O(mlogm+nlogn)。
空间复杂度：O(logm+logn)

其中 m 和 n 分别是两个数组的长度。空间复杂度主要取决于排序使用的额外空间。

function intersect(nums1, nums2) {
    nums1.sort((a, b) => a - b)
    nums2.sort((a, b) => a - b)
    let i1 = 0, i2 = 0, result = []
    while (i1 < nums1.length && i2 < nums2.length) {
        if (nums1[i1] === nums2[i2]) {
            result.push(nums1[i1])
            i1++
            i2++
        }
        if (nums1[i1] < nums2[i2]) {
            i1++
        }
        if (nums1[i1] > nums2[i2]) {
            i2++
        }
    }
    return result
}

console.log('🌰', intersect([4, 9, 5],  [9, 4, 9, 8, 4]))
console.log('🌰', intersect([1, 2, 2, 1], [2, 2]))

执行结果

🌰 [ 4, 9 ]
🌰 [ 2, 2 ]

本页目录