力扣LeetCode，兩個數組的交集 II

1、給定兩個數組，編寫一個函數來計算它們的交集。

示例 1:

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

示例 2:

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

說明：

• 輸出結果中每個元素出現的次數，應與元素在兩個數組中出現的次數一致。
• 我們可以不考慮輸出結果的順序。

進階:

• 如果給定的數組已經排好序呢？你將如何優化你的算法？
• 如果 nums1 的大小比 nums2 小很多，哪種方法更優？
• 如果 nums2 的元素存儲在磁盤上，磁盤內存是有限的，並且你不能一次加載所有的元素到內存中，你該怎麼辦？

2、代碼實現，如下所示：

package com.leetcode;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.TreeMap;

/**
 * @ProjectName: dataConstruct
 * @Package: com.leetcode
 * @ClassName: ArrayIntersection2
 * @Author: biehl
 * @Description: ${description}
 * @Date: 2020/3/15 11:46
 * @Version: 1.0
 */
public class ArrayIntersection2 {

    public int[] intersect(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new TreeMap<Integer, Integer>();
        // 遍歷數組nums1
        for (int num : nums1) {
            // 如果Map集合中不包含此元素
            if (!map.containsKey(num)) {
                // 就將num設置到map中，並將value值設置爲1
                map.put(num, 1);
            } else {
                // 此時，如果map中包含了這個元素
                // 否則，將num的value值加一。
                map.put(num, map.get(num) + 1);
            }
        }

        List<Integer> list = new ArrayList<Integer>();
        // 遍歷數組nums2
        for (int num : nums2) {
            // 判斷，如果map集合中包含了num這個key
            if (map.containsKey(num)) {
                // 就將這個num添加到集合中
                list.add(num);
                // 此時將map集合中的num數字的頻數減一
                map.put(num, map.get(num) - 1);
                // 如果此時num的頻數是0，就刪除掉這個num
                if (map.get(num) == 0) {
                    map.remove(num);
                }
            }
        }

        // 將List集合轉換爲int數組返回
        int[] res = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }

    public static void main(String[] args) {
//        int[] nums1 = new int[]{1, 2, 2, 1};
//        int[] nums2 = {2, 2};
//        int[] nums1 = new int[]{4, 9, 5};
//        int[] nums2 = {9, 4, 9, 8, 4};

        int[] nums1 = new int[]{1, 1, 1, 1, 1, 1};
        int[] nums2 = {1, 1, 1};
        ArrayIntersection2 arrayIntersection = new ArrayIntersection2();
        int[] intersection = arrayIntersection.intersect(nums1, nums2);
        for (int i = 0; i < intersection.length; i++) {
            System.out.println(intersection[i]);
        }
    }

}