leetcode 350、两个数组的交集 II（Map）

给定两个数组，编写一个函数来计算它们的交集。

---

示例 1:

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

示例 2:

输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [4,9]

说明：

• 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
• 我们可以不考虑输出结果的顺序。

---

思路：

用 Map 记录值和每个值出现的频率。

class Solution {
    public int[] intersect(int[] nums1, int[] nums2) {
        TreeMap<Integer, Integer> map = new TreeMap<>();    //后面new的时候不是Map，所以为了避免总出现这个情况，定义的时候先把类型定义好，比如我这里的TreeMap
        for (int i : nums1) {
            if (!map.containsKey(i)) { //不是contains
                map.put(i, 1);
            }
            else {
                map.put(i, map.get(i) + 1);
            }
        }
        
        ArrayList<Integer> list = new ArrayList<>();
        for (int i : nums2) {
            if (map.containsKey(i)) {
                list.add(i);
                map.put(i, map.get(i) - 1); //这里是put,而不是add或set
                if (map.get(i) == 0)
                    map.remove(i);
            }
        }
        
        int[] res = new int [list.size()];
        for (int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }
}