目录

LC 1764. 通过连接另一个数组的子数组得到一个数组 (opens new window) (opens new window)

中等

# 问题描述

给你一个长度为 n 的二维整数数组 groups,同时给你一个整数数组 nums

你是否可以从 nums 中选出 n不相交 的子数组,使得第 i 个子数组与 groups[i](下标从 0 开始)完全相同,且如果 i > 0,那么第 (i-1) 个子数组在 nums 中出现的位置在第 i 个子数组前面。(也就是说,这些子数组在 nums 中出现的顺序需要与 groups 顺序相同)

如果你可以找出这样的 n 个子数组,请你返回 true ,否则返回 false

如果不存在下标为 k 的元素 nums[k] 属于不止一个子数组,就称这些子数组是 不相交 的。子数组指的是原数组中连续元素组成的一个序列。

示例 1:

输入:groups = [[1,-1,-1],[3,-2,0]], nums = [1,-1,0,1,-1,-1,3,-2,0]
输出:true
解释:你可以分别在 nums 中选出第 0 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 和第 1 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 。
这两个子数组是不相交的,因为它们没有任何共同的元素。

示例 2:

输入:groups = [[10,-2],[1,2,3,4]], nums = [1,2,3,4,10,-2]
输出:false
解释:选择子数组 [1,2,3,4,10,-2] 和 [1,2,3,4,10,-2] 是不正确的,因为它们出现的顺序与 groups 中顺序不同。
[10,-2] 必须出现在 [1,2,3,4] 之前。

示例 3:

输入:groups = [[1,2,3],[3,4]], nums = [7,7,1,2,3,4,7,7]
输出:false
解释:选择子数组 [7,7,1,2,3,4,7,7] 和 [7,7,1,2,3,4,7,7] 是不正确的,因为它们不是不相交子数组。
它们有一个共同的元素 nums[4] (下标从 0 开始)。

提示:

  • groups.length == n
  • 1 <= n <= 103
  • 1 <= groups[i].length, sum(groups[i].length) <= 103
  • 1 <= nums.length <= 103
  • -107 <= groups[i][j], nums[k] <= 107

# 贪心

由于这些子数组在 numsnums 中出现的顺序需要与 groupsgroups 顺序相同,因此,遍历数组 numsnums,使用指针 ii 指向当前遍历数组的下标,指针 idxidx 指向当前查找的 groupgroup,若当前的 groupgroupnumsnums 中当前坐标开始后的元素一致,则说明找到满足条件的子数组 idxidx 后移,同时 numsnums 指针也后移到子数组的下一位,最终若 idxidxgroupsgroups 长度一致,则说明能够找出这样的 nn 个子数组,返回 truetrue , 否则返回 falsefalse

/**
 * @param {number[][]} groups
 * @param {number[]} nums
 * @return {boolean}
 */
var canChoose = function (groups, nums) {
  let idx = 0
  const check = (group, start) => {
    if (start + group.length > nums.length) {
      return false
    }
    for (let i = 0; i < group.length; i++) {
      if (group[i] !== nums[start + i]) return false
    }
    return true
  }
  for (let i = 0; i < nums.length && idx < groups.length; ) {
    if (check(groups[idx], i)) {
      i += groups[idx++].length
    } else {
      i++
    }
  }
  return idx === groups.length
}
  • 时间复杂度:O(n×m)O(n \times m)nnnumsnums 数组长度,mmgroupsgroups 中元素的平均长度。
  • 空间复杂度:O(1)O(1)
上次更新: 2023/01/31 19:48:05

本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 , 转载请注明出处!