leetcode-349. 两个数组的交集

JAVA解法

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        // 用于保存 nums1 的 HashSet
        Set<Integer> set1 = new HashSet<>();
        // 用于存储相交元素的 HashSet
        Set<Integer> set2 = new HashSet<>();
        // 遍历 nums1 将其存到 set1 中
        for(int i:nums1){
            set1.add(i);
        }
        // 若 nums2 中存在 set1 中的元素，则加入 set2
        for(int i:nums2){
            if(set1.contains(i)){
                set2.add(i);
            }
        }
        // 开辟一个长度为 set2.size() 的存放相交元素的数组
        int[] arr = new int[set2.size()];
        int j = 0;
        // 把 set2 里的值存到 arr 数组中
        for(int i:set2){
            arr[j++] = i;
        }
        return arr;
    }
}

题解分析

首先想到的是用 Set 来去重存储传进来的数组 nums1，所以遍历 nums1 将所有数组存进去。接下来，遍历 nums2 并加个判断，将与 set1 相同的元素储存到 set2，最后把 set2 存到数组里并返回。

leetcode原题： 349. 两个数组的交集