LeetCode 349. 两个数组的交集

给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是唯一的。我们可以 不考虑输出结果的顺序 。

示例 1：

css 复制代码

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

示例 2：

css 复制代码

输入： 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

这道题目，主要要学会使用一种哈希数据结构：unordered_set，这个数据结构可以解决很多类似的问题。

注意题目特意说明：输出结果中的每个元素一定是唯一的，也就是说输出的结果的去重的，同时可以不考虑输出结果的顺序

这道题用暴力的解法时间复杂度是O(n^2)，那来看看使用哈希法进一步优化。

那么用数组来做哈希表也是不错的选择，例如242. 有效的字母异位词

但是要注意，使用数组来做哈希的题目，是因为题目都限制了数值的大小。

而这道题目没有限制数值的大小，就无法使用数组来做哈希表了。

而且如果哈希值比较少、特别分散、跨度非常大，使用数组就造成空间的极大浪费。

时间复杂度: O(n + m) ，m 是最后要把 set转成vector
空间复杂度: O(n)

那有同学可能问了，遇到哈希问题我直接都用set不就得了，用什么数组啊。

直接使用set 不仅占用空间比数组大，而且速度要比数组慢，set把数值映射到key上都要做hash计算的。

不要小瞧这个耗时，在数据量大的情况，差距是很明显的。

本题后面力扣改了题目描述和后台测试数据，增添了数值范围：

1 <= nums1.length, nums2.length <= 1000
0 <= nums1 $i$ , nums2 $i$ <= 1000

所以就可以使用数组来做哈希表了，因为数组都是 1000以内的。

HashSet 解法：

kotlin 复制代码

fun intersection(nums1: IntArray, nums2: IntArray): IntArray {
    val resultSet = mutableSetOf<Int>()
    val hashSet = hashSetOf<Int>()

    for (i in nums1) {
        hashSet.add(i)
    }

    for (i in nums2) {
        if (hashSet.contains(i)) {
            resultSet.add(i)
        }
    }

    return resultSet.toIntArray()
}

数组解法：

kotlin 复制代码

class Solution {
    fun intersection(nums1: IntArray, nums2: IntArray): IntArray {
        val result = mutableListOf<Int>()
        val array1 = Array(size = 1002) { 0 }
        val array2 = Array(size = 1002) { 0 }
        for (i in nums1) {
            array1[i]++
        }

        for (i in nums2) {
            array2[i]++
        }

        for (i in 0..<1002) {
            if (array1[i] > 0 && array2[i] > 0) {
                result.add(i)
            }
        }
        return result.toIntArray()
    }
}