【精】LintCode领扣算法问题答案：548. 两数组的交集 II

548. 两数组的交集 II

描述

给定两个数组，计算两个数组的交集

• 每个元素出现次数得和在数组里一样
• 答案可以以任意顺序给出

样例 1：

输入: 
	nums1 = [1, 2, 2, 1], nums2 = [2, 2]
输出: 
	[2, 2]

样例 2：

输入: 
	nums1 = [1, 1, 2], nums2 = [1]
输出: 
	[1]

挑战

-如果给定的数组已经排序了怎么办?如何优化算法?
-如果nums1的大小比num2的小怎么办?哪种算法更好?
-如果nums2的元素存储在磁盘上，并且内存受到限制，以至于不能一次将所有元素加载到内存中，该怎么办?

原题传送门

---

---

题解

public class Solution {
    /**
     * @param nums1: an integer array
     * @param nums2: an integer array
     * @return: an integer array
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        // write your code here

        Map<Integer, Integer> counter1 = new HashMap<>();
        Map<Integer, Integer> counter2 = new HashMap<>();
        
        for (int num : nums1) {
            counter1.put(num, counter1.getOrDefault(num, 0) + 1);
        }

        for (int num : nums2) {
            counter2.put(num, counter2.getOrDefault(num, 0) + 1);
        }
        
        List<Integer> list = new ArrayList<>();
        for (Map.Entry<Integer, Integer> row : counter1.entrySet()) {
            Integer count2 = counter2.get(row.getKey());
            if (count2 != null) {
                for (int i = 0; i < Math.min(row.getValue(), count2); ++i) {
                    list.add(row.getKey());
                }
            }
        }
        
        int[] ans = new int[list.size()];
        {
            int i = 0;
            for (Integer num : list) {
                ans[i++] = num;
            }
        }
        
        return ans;
    }
}

最后说两句

非常感谢你阅读本文章，如果你觉得本文对你有所帮助，请留下你的足迹，点个赞，留个言，多谢~

作者水平有限，如果文章内容有不准确的地方，请指正。

希望小伙伴们都能每天进步一点点。

声明

本文由二当家的白帽子博客原创，转载请注明来源，谢谢~