LeetCode.349. 两个数组的交集

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

例子:

给定 num1= [1, 2, 2, 1], nums2 = [2, 2], 返回 [2].

提示:

  1. 每个在结果中的元素必定是唯一的。
  2. 我们可以不考虑输出结果的顺序。

分析:
先考虑去重,使用 Set 的特性,无重复元素,把两个数组存进 Set,
所谓并集,就是第一个数组有,第二个数组也有,遍历第一个 Set,如果第二个 Set 中也有那么该元素就是并集元素,
最后剪切数组

class Solution {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        int[] ans = new int[nums1.length + nums2.length];
        int k = 0;
        for (int i = 0; i < nums1.length; i++)
            set1.add(nums1[i]);
        for (int i = 0; i < nums2.length; i++)
            set2.add(nums2[i]);
        for (Integer integer : set1)
            if (set2.contains(integer))
                ans[k++] = integer;
        return Arrays.copyOfRange(ans, 0, k);
    }
}

猜你喜欢

转载自blog.csdn.net/liyuanyue2017/article/details/81200256