Question

2248. 多个数组求交集

English Version

题目描述

给你一个二维整数数组 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] 中的所有值 互不相同

解法

方法一：计数

遍历数组 nums，对于每个数组 arr，统计数组 arr 中每个数字出现的次数，然后遍历计数数组，统计出现次数等于数组 nums 的长度的数字，即为答案。

时间复杂度 $O(N)$，空间复杂度 $O(1000)$。其中 $N$ 为数组 nums 中数字的总数。

class Solution:
  def intersection(self, nums: List[List[int]]) -> List[int]:
    cnt = [0] * 1001
    for arr in nums:
      for x in arr:
        cnt[x] += 1
    return [x for x, v in enumerate(cnt) if v == len(nums)]

class Solution {
  public List<Integer> intersection(int[][] nums) {
    int[] cnt = new int[1001];
    for (var arr : nums) {
      for (int x : arr) {
        ++cnt[x];
      }
    }
    List<Integer> ans = new ArrayList<>();
    for (int x = 0; x < 1001; ++x) {
      if (cnt[x] == nums.length) {
        ans.add(x);
      }
    }
    return ans;
  }
}

class Solution {
public:
  vector<int> intersection(vector<vector<int>>& nums) {
    int cnt[1001]{};
    for (auto& arr : nums) {
      for (int& x : arr) {
        ++cnt[x];
      }
    }
    vector<int> ans;
    for (int x = 0; x < 1001; ++x) {
      if (cnt[x] == nums.size()) {
        ans.push_back(x);
      }
    }
    return ans;
  }
};

func intersection(nums [][]int) (ans []int) {
  cnt := [1001]int{}
  for _, arr := range nums {
    for _, x := range arr {
      cnt[x]++
    }
  }
  for x, v := range cnt {
    if v == len(nums) {
      ans = append(ans, x)
    }
  }
  return
}

function intersection(nums: number[][]): number[] {
  const cnt = new Array(1001).fill(0);
  for (const arr of nums) {
    for (const x of arr) {
      cnt[x]++;
    }
  }
  const ans: number[] = [];
  for (let x = 0; x < 1001; x++) {
    if (cnt[x] === nums.length) {
      ans.push(x);
    }
  }
  return ans;
}

class Solution {
  /**
   * @param Integer[][] $nums
   * @return Integer[]
   */
  function intersection($nums) {
    $rs = [];
    for ($i = 0; $i < count($nums); $i++) {
      for ($j = 0; $j < count($nums[$i]); $j++) {
        $hashtable[$nums[$i][$j]] += 1;
        if ($hashtable[$nums[$i][$j]] === count($nums)) {
          array_push($rs, $nums[$i][$j]);
        }
      }
    }
    sort($rs);
    return $rs;
  }
}

方法二

class Solution:
  def intersection(self, nums: List[List[int]]) -> List[int]:
    cnt = Counter()
    ans = []
    for arr in nums:
      for x in arr:
        cnt[x] += 1
        if cnt[x] == len(nums):
          ans.append(x)
    ans.sort()
    return ans

class Solution {
  public List<Integer> intersection(int[][] nums) {
    Map<Integer, Integer> cnt = new HashMap<>();
    List<Integer> ans = new ArrayList<>();
    for (var arr : nums) {
      for (int x : arr) {
        if (cnt.merge(x, 1, Integer::sum) == nums.length) {
          ans.add(x);
        }
      }
    }
    Collections.sort(ans);
    return ans;
  }
}

class Solution {
public:
  vector<int> intersection(vector<vector<int>>& nums) {
    unordered_map<int, int> cnt;
    vector<int> ans;
    for (auto& arr : nums) {
      for (int& x : arr) {
        if (++cnt[x] == nums.size()) {
          ans.push_back(x);
        }
      }
    }
    sort(ans.begin(), ans.end());
    return ans;
  }
};

func intersection(nums [][]int) (ans []int) {
  cnt := map[int]int{}
  for _, arr := range nums {
    for _, x := range arr {
      cnt[x]++
      if cnt[x] == len(nums) {
        ans = append(ans, x)
      }
    }
  }
  sort.Ints(ans)
  return
}

function intersection(nums: number[][]): number[] {
  const cnt = new Array(1001).fill(0);
  const ans: number[] = [];
  for (const arr of nums) {
    for (const x of arr) {
      if (++cnt[x] == nums.length) {
        ans.push(x);
      }
    }
  }
  ans.sort((a, b) => a - b);
  return ans;
}