[Leetcode] 350. Intersection of Two Arrays II

最新推荐文章于 2025-02-11 22:00:00 发布

原创最新推荐文章于 2025-02-11 22:00:00 发布 · 83 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 专栏收录该内容

70 篇文章

订阅专栏

本文详细解析了LeetCode上两数组交集II问题的两种高效解决方案：使用哈希映射和排序方法。通过实例代码展示了如何利用Java的HashMap进行元素计数，以及如何通过双指针技巧在排序后的数组中寻找交集。

Problems: https://leetcode.com/problems/intersection-of-two-arrays-ii/

Solution1:
Using map

只需要存储nums1的map
对于输出，不需要另外创建数组，而是直接重写nums1

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        // using map
        Map<Integer, Integer> map = new HashMap<>();
        for(int i : nums1) {
            map.put(i, map.getOrDefault(i, 0) + 1);
        }
        
        int k = 0;
        // 将nums1存完后直接用nums2中的值作为key进行查找
        for(int i : nums2) {
            if(map.containsKey(i) && map.get(i) > 0) {
            // 重写nums1
                nums1[k++] = i;
                // 注意找到以后key的value要-1
                map.put(i, map.get(i) - 1);
            }
        }
        
        return Arrays.copyOf(nums1, k);
    }
}

Solution2:
Using sort

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        // sort
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i = 0, j = 0, k = 0;
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                i++;
            } else if (nums1[i] > nums2[j]) {
                j++;
            } else {
                nums1[k++] = nums1[i];
                // 注意此时idx i 和 j 都要+1
                i++;
                j++;
            }
        }
        return Arrays.copyOf(nums1, k);
    }
}