题目链接
https://leetcode-cn.com/problems/intersection-of-two-arrays-ii/
描述
给定两个数组,编写一个函数来计算它们的交集。
说明:
输出结果中每个元素出现的次数,应与元素在两个数组中出现次数的最小值一致。
我们可以不考虑输出结果的顺序。
进阶:
如果给定的数组已经排好序呢?你将如何优化你的算法?
如果 nums1 的大小比 nums2 小很多,哪种方法更优?
如果 nums2 的元素存储在磁盘上,内存是有限的,并且你不能一次加载所有的元素到内存中,你该怎么办?
示例
示例 1:
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例 2:
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
初始代码模板
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
}
}
代码
基本的问题比较简单,进阶问题可以去看看链接里的官方题解。
class Solution {
public int[] intersect(int[] nums1, int[] nums2) {
List<Integer> res = new LinkedList<>();
Map<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums1.length; i++) {
map.put(nums1[i], map.getOrDefault(nums1[i], 0) + 1);
}
for (int i = 0; i < nums2.length; i++) {
if (map.containsKey(nums2[i])) {
res.add(nums2[i]);
if (map.get(nums2[i]) == 1) {
map.remove(nums2[i]);
} else {
map.put(nums2[i], map.get(nums2[i]) - 1);
}
}
}
int[] arr = new int[res.size()];
for (int i = 0; i < arr.length; i++) {
arr[i] = res.get(i);
}
return arr;
}
}