力扣Java - 350. 两个数组的交集 II
两个数组的交集
- 题目描述
- 实例演示
-
-
-
- 示例一
- 示例二
-
-
- 算法思想
- 算法代码
题目描述
- 两个整数数组 nums1 和 nums2
- 要求:以数组形式返回两数组的交集。
- 返回结果中每个元素出现的次数,应与元素在两个数组中都出现的次数一致(如果出现次数不一致,则考虑取较小值)。
- 可以不考虑输出结果的顺序。
实例演示
示例一
输入:nums1 = [1,2,2,1], nums2 = [2,2]
输出:[2,2]
示例二
输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[4,9]
算法思想
- 借助 HashMap 和 ArrayList 实现。
- 使用增强 for 循环将 nums1 数组存入 HashMap 集合中,其中 key 存放数组元素值,value 存放该值出现的次数。key 是不可重复的。
- 然后依次遍历 nums 数组中的元素,与 HashMap 中的 key 比较,看是否包含,如果包含且该 key 对应的 value 值不等于 0 ,则将该数组元素添加到 ArrayList 集合中,同时 HashMap 中该 key 对应的 value 值 减1。否则什么也不需要做。
- nums2 数组全部遍历完之后,ArrayList 集合中的元素就是所求,将其赋值给数组返回即可。
算法代码
class Solution { public int[] intersect(int[] nums1, int[] nums2) { HashMap<Integer, Integer> map = new HashMap<Integer, Integer>(); ArrayList<Integer> list = new ArrayList<Integer>(); for (int num : nums1){ int count = (map.containsKey(num) ? map.get(num) : 0) + 1; map.put(num,count); } for (int i = 0; i < nums2.length; i++){ if (map.containsKey(nums2[i]) && map.get(nums2[i]) != 0){ list.add(nums2[i]); int num = map.get(nums2[i]); map.replace(nums2[i],num - 1); } } int[] ret = new int[list.size()]; Iterator<Integer> iterator = list.iterator(); int index = 0; while(iterator.hasNext()){ ret[index++] = iterator.next(); } return ret; }}