两个数组的交集（LeetCode 349）

最新推荐文章于 2024-11-10 14:06:08 发布

pepsi_w

最新推荐文章于 2024-11-10 14:06:08 发布

阅读量262

点赞数

分类专栏：小菜鸡的JAVA学习文章标签： leetcode 算法 hash

本文链接：https://blog.csdn.net/wangyumei0916/article/details/131465066

版权

小菜鸡的JAVA学习专栏收录该内容

22 篇文章 0 订阅

订阅专栏

该文章介绍了一种使用HashSet解决求两个整数数组交集的Java算法。首先，确保一个数组比另一个长，然后将长数组的元素存入HashSet。接着遍历短数组，尝试将每个元素添加到HashSet中，添加失败则说明是交集并添加到结果集合。最后，将结果集合转换回int数组返回。

摘要由CSDN通过智能技术生成

题目

349. 两个数组的交集

思路

将较长的序列放入一个set后，再加入短序列的数字，判断当前数字是否添加成功，如果添加成功则表示set中没有该数字，则不属于两个数组之间的交集，将该数字从set中移除（移除是因为保证set的纯洁性即只含有长序列中的数字）；如果添加失败，则表示该数字则是两个数组的交集，将该数字添加到交集set中。最后由于该函数返回的是int[]，则需要将交集set转化为int[].

代码

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
         HashSet<Integer> temp_1 = new HashSet<Integer>();
        HashSet<Integer> temp_2 = new HashSet<Integer>();
//        保证nums1是较长的序列
        if(nums1.length < nums2.length){
            int[] temp = nums1;
            nums1 = nums2;
            nums2 = temp;
            
        }
        for (int i: nums1) {
            temp_1.add(i);
        }
        for (int i: nums2) {
            // 添加成功 表示set中没有该数字，则不是交集
            if(!temp_1.add(i)){
                temp_2.add(i);
            }
            else{
                temp_1.remove(i);
            }
        }
        // 将set转化为int[]
        // 可以使用该语句进行代替：return temp_2.stream().mapToInt(x -> x).toArray();
        int[] out = new int[temp_2.size()];
        int i = 0;
        for (int a: temp_2) {
            out[i] = a;
            i++;
        }
        return out;
    }
}