[LeetCode] 349. 两个数组的交集

1 题目描述

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

示例 1:

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

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]
说明:

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2 解题思路

看到这道题的想法就是 设置两个去重之后的set，然后短的那个去长的那个里面查找，
看到题解之后 其实是不太明白的 题解的思路我也没有看太懂
然后翻了一下评论区， 很喜欢这个做法 觉得比较是个小白上手

 假设长度小的数组为 less, 长度为 M；长度大的数组为 more, 长度为 N

使用长度小的数组建立哈希表 lessSet，用于匹配查找，时间复杂度为 O(M), 最大额外空间复杂度为 O(M).
循环长度大的数组，通过哈希表 lessSet 匹配与长度小的数组的交集，查找时间复杂度为 O(1)，总的时间复杂度为 O(N).
总体的时间复杂度为 O(M+N), 额外空间复杂度为 O(M+N).

3 解决代码

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        int len1 = nums1.length;
        int len2 = nums2.length;
        int[] less = len1 < len2 ? nums1:nums2;
        int[] more = len1 < len2 ? nums2:nums1;
        //长度小的建立哈希表，用于匹配查找
        HashSet<Integer> lessSet = new HashSet<Integer>();
        for(int i = 0,len = less.length;i<len;i++){
            lessSet.add(less[i]);
        }
        //长度大的数组进行查找，与哈希表进行匹配
        HashSet<Integer> moreSet = new HashSet<Integer>();
        for(int i = 0,len = more.length;i<len;i++){
            if(!moreSet.contains(more[i]) && lessSet.contains(more[i])){
                moreSet.add(more[i]);
            }
        }
        //Set 转为 数组
        int size = moreSet.size(),  k = 0;
        int[] ans = new int[size];
        for(int n:moreSet){
            ans[k++] = n;
        }
        return ans;
        
        
    }
}