leetCode: 350. 两个数组的交集 II Java映射实现

最新推荐文章于 2022-10-27 20:14:39 发布

戴眼镜的小青年

最新推荐文章于 2022-10-27 20:14:39 发布

阅读量274

点赞数

CC 4.0 BY-SA版权

分类专栏： Java 数据结构数据结构篇文章标签： LeetCode Java 映射算法

本文链接：https://blog.youkuaiyun.com/weixin_44594056/article/details/94290433

Java 同时被 3 个专栏收录

12 篇文章

订阅专栏

数据结构篇

10 篇文章

订阅专栏

数据结构

8 篇文章

订阅专栏

本文介绍了一种使用TreeMap实现两个整数数组交集的算法。通过遍历第一个数组，将元素及其出现次数存储在TreeMap中，然后遍历第二个数组，检查元素是否存在于TreeMap中。如果存在，则将该元素添加到结果列表中，并更新TreeMap中该元素的计数。最终将结果列表转换为数组返回。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

实现代码如下:

public int[] intersect(int[] nums1, int[] nums2) {
        TreeMap<Integer, Integer> map = new TreeMap<>();
        for(int num : nums1){
            if(!map.containsKey(num))
               map.put(num,1);
            else
               map.put(num, map.get(num) + 1);
        }
        ArrayList<Integer>list = new ArrayList<>();
        for(int num : nums2){
            if(map.containsKey(num)){
                list.add(num);
                map.put(num, map.get(num) - 1);
                if(map.get(num) == 0)
                    map.remove(num);
            }  
        }
        int []res = new int[list.size()];
        for(int i = 0; i < list.size() ; i++)
               res[i] = list.get(i);
        return res;
    }