【Map】【List】【数组】获得两个数组的交集

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

示例 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