力扣LeetCode，两个数组的交集 II

 1 package com.leetcode;
 2 
 3 import java.util.ArrayList;
 4 import java.util.List;
 5 import java.util.Map;
 6 import java.util.TreeMap;
 7 
 8 /**
 9  * @ProjectName: dataConstruct
10  * @Package: com.leetcode
11  * @ClassName: ArrayIntersection2
12  * @Author: biehl
13  * @Description: ${description}
14  * @Date: 2020/3/15 11:46
15  * @Version: 1.0
16  */
17 public class ArrayIntersection2 {
18 
19     public int[] intersect(int[] nums1, int[] nums2) {
20         Map<Integer, Integer> map = new TreeMap<Integer, Integer>();
21         // 遍历数组nums1
22         for (int num : nums1) {
23             // 如果Map集合中不包含此元素
24             if (!map.containsKey(num)) {
25                 // 就将num设置到map中，并将value值设置为1
26                 map.put(num, 1);
27             } else {
28                 // 此时，如果map中包含了这个元素
29                 // 否则，将num的value值加一。
30                 map.put(num, map.get(num) + 1);
31             }
32         }
33 
34         List<Integer> list = new ArrayList<Integer>();
35         // 遍历数组nums2
36         for (int num : nums2) {
37             // 判断，如果map集合中包含了num这个key
38             if (map.containsKey(num)) {
39                 // 就将这个num添加到集合中
40                 list.add(num);
41                 // 此时将map集合中的num数字的频数减一
42                 map.put(num, map.get(num) - 1);
43                 // 如果此时num的频数是0，就删除掉这个num
44                 if (map.get(num) == 0) {
45                     map.remove(num);
46                 }
47             }
48         }
49 
50         // 将List集合转换为int数组返回
51         int[] res = new int[list.size()];
52         for (int i = 0; i < list.size(); i++) {
53             res[i] = list.get(i);
54         }
55         return res;
56     }
57 
58     public static void main(String[] args) {
59 //        int[] nums1 = new int[]{1, 2, 2, 1};
60 //        int[] nums2 = {2, 2};
61 //        int[] nums1 = new int[]{4, 9, 5};
62 //        int[] nums2 = {9, 4, 9, 8, 4};
63 
64         int[] nums1 = new int[]{1, 1, 1, 1, 1, 1};
65         int[] nums2 = {1, 1, 1};
66         ArrayIntersection2 arrayIntersection = new ArrayIntersection2();
67         int[] intersection = arrayIntersection.intersect(nums1, nums2);
68         for (int i = 0; i < intersection.length; i++) {
69             System.out.println(intersection[i]);
70         }
71     }
72 
73 }