> 文档中心 > 力扣Java - 350. 两个数组的交集 II

力扣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;    }}