LeetCode 349. 两个数组的交集

目录结构

1.题目

2.题解

2.1集合Set

2.2内置函数retainAll

---

1.题目

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

示例：

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


输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
输出：[9,4]

说明：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/intersection-of-two-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

2.1集合Set

使用集合去重，且实现O(1)的add和contains。

public class Solution349 {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set = new HashSet<>();
        Set<Integer> set1 = new HashSet<>();
        for (int n : nums1) {
            set1.add(n);
        }
        for (int n : nums2) {
            if (set1.contains(n)) {
                set.add(n);
            }
        }
        int[] result = new int[set.size()];
        int p = 0;
        for (int n : set) {
            result[p++] = n;
        }
        return result;
    }
}

• 时间复杂度：
• 空间复杂度：

2.2内置函数retainAll

public class Solution349 {
    public int[] intersection(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        for (int n : nums1) {
            set1.add(n);
        }
        for (int n : nums2) {
            set2.add(n);
        }
        set1.retainAll(set2);
        int[] result = new int[set1.size()];
        int p = 0;
        for (int n : set1) {
            result[p++] = n;
        }
        return result;
    }
}

• 时间复杂度：或
• 空间复杂度：