映射 04 Leetcode中的350号问题

原创于 2019-02-11 23:26:00 发布 · 60 阅读

0 ·

CC 4.0 BY-SA版权

数据结构 [Java] 专栏收录该内容

86 篇文章

订阅专栏

本文详细解析了LeetCode上350题两个数组的交集II的算法实现，通过使用TreeMap记录第一个数组中元素的出现次数，然后遍历第二个数组，将存在于第一个数组中的元素加入结果列表，并更新元素在TreeMap中的计数，直至遍历完成。该算法巧妙地利用了TreeMap的有序性和高效查找特性，实现了两个数组交集的有效求解。

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

350.两个数组的交集 II

class Solution { 

    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 < res.length; i++) {
            res[i] = list.get(i);
        }
        return res;
    }

}