349. 两个数组的交集——学习笔记

发布时间 2023-04-12 22:27:55作者: 会飞的笨笨

题目:给定两个数组 nums1 和 nums2 ,返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。

示例1

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

示例2

输入:nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出:[9,4]
解释:[4,9] 也是可通过的

提示

  • 1 <= nums1.length, nums2.length <= 1000
  • 0 <= nums1[i], nums2[i] <= 1000

题目来源:力扣(LeetCode)链接

题解

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        //1.创建set1用来给nums1中的元素去重
        Set<Integer> set1 = new HashSet<>();
        for (int i = 0; i < nums1.length; i++) {
            set1.add(nums1[i]);
        }
        //2.创建set2存放两个数组的交集,并去出重复部分
        Set<Integer> set2 = new HashSet<>();
        for (int i = 0; i < nums2.length; i++) {
            //3.遍历nums2中的每一个元素,利用HashSet的contains()方法判断是否存在交集
            if (set1.contains(nums2[i])) {
                set2.add(nums2[i]);
            }
        }
        //set2中存放的就是两数组的交集
        int[] res = new int[set2.size()];
        int i = 0;
        //4.创建数组,遍历set2,把元素放入数组中返回
        for (Integer num : set2) {
            res[i++] = num;
        }
        return res;
    }
}