Java 力扣-----两个数组的交集

1.题目描述

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

2.题目示例

3.思路及代码

• 思路：通过set去重，list往里加，最后转为数组即可
• 代码：

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        for(int i : nums1){
            set.add(i);
        }
        List<Integer> list = new ArrayList<>();
        for(int num : nums2){
            if(set.contains(num)){
                list.add(num);
                //去掉set中该元素，保证唯一性
                set.remove(num);
            }
        }
        int[] res = new int[list.size()];
        for(int i = 0; i < list.size(); i++){
            res[i] = list.get(i);
        }
        return res;
    }
}