一起刷Leetcode 349-350两个数组的交集

1. 第349题

1. 题目简介

解题模板：

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {

    }
}

1.2 题目分析

需要注意的是计算出交集是不能包含重复元素，所以可以考虑先把交集求出来，然后用set集合过滤。这种方法比较笨，也比较好理解。

• 方法： 先把nums1过滤掉重复元素即转入set中，然后遍历nums2，判断set中是否包含该元素，如果有，该元素就是重复元素。
  代码如下：

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {

        Set<Integer> set = new HashSet<>();
        for (int num:nums1){
            set.add(num);
        }
        ArrayList<Integer> list = new ArrayList<>();
        for (int num : nums2){
            if (set.contains(num)){ //如果set中包含num
                list.add(num);
                set.remove(num);//因为list中不能有重复元素，所以需要remove一下
            }
        }

        int [] arr = new int[list.size()];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = list.get(i);
        }
        return arr;
    }
}

2. 第350题

2.1 题目简介

2.1 题目分析

注意，交集中可以出现重复元素，所以不能用使用set集合装数组中元素，这里用map数据结构来装，将第一个数组转入map中，key装表元素值，value装对应元素出现的次数。接着遍历第二个数组，数组中元素在map的key中，那么这个元素就是交集中的一个元素，将该元素取出，value值减一即可。具体实现代码如下：

import java.util.ArrayList;
import java.util.TreeMap;

public class Solution350 {
    public int[] intersection(int[] nums1, int[] nums2) {

        TreeMap<Integer,Integer> map = new TreeMap<>(); // 定义一个map，键为元素，值为元素出现次数
        // 统计每个元素出现的次数
        for (int num:nums1){
            if (!map.containsKey(num)) {
                map.put(num, 1);
            }else {
                map.put(num,map.get(num) + 1);
            }
        }

        ArrayList<Integer> list = new ArrayList<>();
        // 遍历第二个数组
        for (int num : nums2){
            if (map.containsKey(num)) { // 如果map的key包含了num，就证明这个num是交集
                list.add(num);      // 将交集元素添加到list中
                map.put(num,map.get(num) - 1);  // 添加完成后将元素出现的次数减一
                if (map.get(num) == 0){
                    map.remove(num);
                }
            }
        }
        int[] arr = new int[list.size()];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = list.get(i);
        }
        return arr;
    }
}