力扣2248:多个数组求交集(Java多种方法)

148 阅读1分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 26 天,点击查看活动详情

一、题目描述

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

示例 1:

输入: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:

输入: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] 中的所有值 互不相同

二、思路讲解

2.1 方法一:统计数字出现次数

因为nums[i]中所有的值互不相同,所以我们可以统计数组在nums中出现过的次数,如果出现次数和nums的长度相同,说明是交集中的值

class Solution {
    public List<Integer> intersection(int[][] nums) {
        List<Integer> list = new ArrayList<>();
        Map<Integer, Integer> map = new HashMap<>();
        for(int i=0; i<nums.length; i++) {
            for(int j=0; j<nums[i].length; j++) {
                map.put(nums[i][j], map.getOrDefault(nums[i][j], 0) + 1);
                if(map.get(nums[i][j]) == nums.length) {
                    list.add(nums[i][j]);
                }
            }
        }
        Collections.sort(list);
        return list;
    }
}

2.2 方法二:模拟

模拟集合求交集的方式,定义哈希集合res,并将第一个集合中的值放入res中,若后面的数组中的数在res中,则放入temp集合中,然后res = temp。res就是前i个数组的交集。

class Solution {
    public List<Integer> intersection(int[][] nums) {
        //存储交集
        Set<Integer> res = new HashSet<>(); 
 
        //将第一个数组的值赋给set
        for(int i=0; i<nums[0].length; i++) {
            res.add(nums[0][i]);
        }
 
        //求前i个数组的交集
        for(int i=1; i<nums.length; i++) {
            Set<Integer> temp = new HashSet<>();
            for(int j=0; j<nums[i].length; j++) {
                if(res.contains(nums[i][j])) {
                    temp.add(nums[i][j]);
                }
            }
            res = temp;
        }
        List<Integer> list = new ArrayList(res);
        Collections.sort(list);
        return list;
    }
}