两个数组的交集

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

示例 1:

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

示例 2:

1
2
输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出: [9,4]

说明:

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

代码

利用两个Set来过滤掉重复出现的元素

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
public int[] intersection(int[] nums1, int[] nums2) {
    if (nums1 == null || nums2 == null || nums1.length == 0 || nums2.length == 0) return new int[0];
    HashSet<Integer> set1 = new HashSet<>();
    HashSet<Integer> resultSet = new HashSet<>();
    for (int i : nums1) {
        //1.去除掉nums1数组中的重复元素
        set1.add(i);
    }
    for (int i : nums2) {
        if (set1.contains(i)) {
            //2.去除掉nums2数组中重复的元素
            resultSet.add(i);
        }
    }
    int[] result = new int[resultSet.size()];
    Iterator<Integer> iterator = resultSet.iterator();
    int index = 0;
    while (iterator.hasNext()) {
        result[index++] = iterator.next();
    }
    return result;
}