給定兩個陣列，編寫一個函式來計算它們的交集。

示例 1:

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

示例 2:

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

輸出: [2]

說明：

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

package com.dylan.practice.leetcode;

import java.util.*;

/**
 * 兩個陣列交集
 *
 * @author xusucheng
 * @create 2018-10-18
 **/
public class Intersect {
    public static int[] intersect(int[] nums1, int[] nums2) {
        Map<Integer, Integer> counter = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        //統計每個元素出現個數
        for (int i = 0; i < nums1.length; i++) {
            int num = nums1[i];
            if (counter.containsKey(num)) {
                counter.put(num, counter.get(num) + 1);
            } else {
                counter.put(num, 1);
            }
        }
        //在第2個數組中查詢，如果找到相同元素放入list中，同時計數減一
        for (int i = 0; i < nums2.length; i++) {
            int num = nums2[i];
            if(counter.containsKey(num) && counter.get(num)>0){
                counter.put(num, counter.get(num)-1);
                list.add(num);
            }
        }

        int[] nums = new int[list.size()];

        for (int i = 0; i < list.size(); i++) {
            nums[i] = list.get(i);
        }

        return nums;
    }

    public static void main(String[] args) {
        int[] nums1 = {1, 2, 2, 1};
        int[] nums2 = {2};
        int[] nums = intersect(nums1, nums2);
        for (int i = 0; i < nums.length; i++) {
            System.out.println(nums[i]);
        }

    }
}
 

輸出：

2