2248. 多个数组求交集

Posted on | In
Words count in article: 319 | Reading time ≈ 1

题目

给你一个二维整数数组 nums ，其中 nums[i] 是由 不同 正整数组成的一个非空数组，按 升序排列 返回一个数组，数组中的每个元素在 nums 所有数组 中都出现过。

示例1：

1
2
3
4
输入：nums = [[3,1,2,4,5],[1,2,3,4],[3,4,5,6]]
输出：[3,4]
解释：
nums[0] = [3,1,2,4,5]，nums[1] = [1,2,3,4]，nums[2] = [3,4,5,6]，在 nums 中每个数组中都出现的数字是 3 和 4 ，所以返回 [3,4] 。

示例2：

1
2
3
4
输入：nums = [[1,2,3],[4,5,6]]
输出：[]
解释：
不存在同时出现在 nums[0] 和 nums[1] 的整数，所以返回一个空列表 [] 。

提示：

  • 1 <= nums.length <= 1000
  • 1 <= sum(nums[i].length) <= 1000
  • 1 <= nums[i][j] <= 1000
  • nums[i] 中的所有值 互不相同

解法

解法一：

Java

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
public List<Integer> intersection(int[][] nums) {
        if (Objects.isNull(nums) || 0 == nums.length) {
            return Collections.emptyList();
        }
        Set<Integer> set = new HashSet<>(nums[0].length);
        
        for (int num : nums[0]) {
            set.add(num);
        }
        
        for (int i = 1;i < nums.length;i++) {
            Set<Integer> newSet = new HashSet<>(nums[i].length);
            for (int j = 0;j < nums[i].length;j++) {
                newSet.add(nums[i][j]);
            }
            set.retainAll(newSet);
        }
        List<Integer> result = new ArrayList<>(set);
        result.sort((Comparator.comparingInt(o -> o)));
        return result;
    }
0%