两个数组的交集

题目

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

示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

• 暴力遍历,可以将两个数组嵌套遍历,将重复数字存入set集合.最后转为int[]数组返回.但是这样时间复杂度会很高
• 将其中一个数组存入set集合,然后依次按另一个数组get,get到的放入结果set中.此处使用stream的和使用遍历差距很大.

代码

• 使用流式处理,耗时78ms

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        if(nums1.length == 0 || nums2.length == 0){
            return new int[]{};
        }
        HashSet<Integer> set = new HashSet<>();
        for(int i = 0;i < nums1.length;i++){
            set.add(nums1[i]);
        }
        HashSet<Integer> resultSet = new HashSet<>();
        for(int i = 0;i < nums2.length;i++){
            if(set.contains(nums2[i])){
                resultSet.add(nums2[i]);
            }
        }
        return resultSet.stream().mapToInt(Number::intValue).toArray();
    }
}

• 使用遍历处理.耗时7ms

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
                //
        if(nums1.length == 0 || nums2.length == 0){
            return new int[]{};
        }
        HashSet<Integer> set = new HashSet<>();
        for(int i = 0;i < nums1.length;i++){
            set.add(nums1[i]);
        }
        HashSet<Integer> resultSet = new HashSet<>();
        for(int i = 0;i < nums2.length;i++){
            if(set.contains(nums2[i])){
                resultSet.add(nums2[i]);
            }
        }
        int[] result = new int[resultSet.size()];
        int i = 0;
        for (Integer integer : resultSet) {
            result[i++] = integer;
        }
        return result;
    }
}