能否连接形成数组

90 阅读1分钟

这是我参与「掘金日新计划 · 2 月更文挑战」的第 6 天,点击查看活动详情

问题描述

给你一个整数数组 arr ,数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces,其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是,不允许 对每个数组 pieces[i] 中的整数重新排序。

如果可以连接 pieces 中的数组形成 arr ,返回 true ;否则,返回 false 。

示例 1:

输入:arr = [15,88], pieces = [[88],[15]]
输出:true
解释:依次连接 [15] 和 [88]

示例 2:

输入:arr = [49,18,16], pieces = [[16,18,49]]
输出:false
解释:即便数字相符,也不能重新排列 pieces[0]

示例 3:

输入:arr = [91,4,64,78], pieces = [[78],[4,64],[91]]
输出:true
解释:依次连接 [91][4,64][78]

提示:

  • 1 <= pieces.length <= arr.length <= 100
  • sum(pieces[i].length) == arr.length
  • 1 <= pieces[i].length <= arr.length
  • 1 <= arr[i], pieces[i][j] <= 100
  • arr 中的整数 互不相同
  • pieces 中的整数 互不相同(也就是说,如果将 pieces 扁平化成一维数组,数组中的所有整数互不相同)

思路分析

首先我们先要理解一下题目意思,题目会给我们一个整数数组 arr ,数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces,其中的整数也 互不相同 。我们需要将 pieces 中的数组按任意顺序连接得到 arr 数组,但不允许 对每个数组 pieces[i] 中的整数重新排序。也就是说我们可以调整 pieces 中子数组的顺序,但不能调整子数组中每个整数的位置。

提示中说到:arr 中的整数 互不相同pieces 中的整数 互不相同(也就是说,如果将 pieces 扁平化成一维数组,数组中的所有整数互不相同),通过这两个提示我们可以知道数组中每一个整数对应的下标都是各不相同的,所以我们可以根据每个整数在 arr 数组中出现的下标来对 pieces 数组中的子数组进行排序。

  • 1、对 pieces 数组进行排序

因为我们不能调整 pieces 中子数组的每个整数的位置,所以我们可以直接根据子数组的第一个元素在 arr 数组中的相对位置来进行排序,具体代码如下:

pieces.sort((a, b) => arr.indexOf(a[0]) - arr.indexOf(b[0]));
  • 2、按顺序对比每个元素

(1)可以将排好序的 pieces 进行扁平化再与 arr 数据进行比较

const arr1 = [];
for (const p of pieces) arr1.push(...p);
for (let i = 0; i < arr.length; i++) {
  if (arr[i] != arr1[i]) return false;
}
return arr.length == arr1.length;

(2)我们也可以直接遍历排序后的数组,按顺序匹配每一项是否相同。

let index = 0;
for (const p of pieces) {
  for (let i = 0; i < p.length; i++) {
    if (arr[index++] != p[i]) return false;
  }
}
return true;

完整 AC 代码如下:

AC 代码

/**
 * @param {number[]} arr
 * @param {number[][]} pieces
 * @return {boolean}
 */
var canFormArray = function (arr, pieces) {
  pieces.sort((a, b) => arr.indexOf(a[0]) - arr.indexOf(b[0]));
  let index = 0;
  for (const p of pieces) {
    for (let i = 0; i < p.length; i++) {
      if (arr[index++] != p[i]) return false;
    }
  }
  return true;
};

说在后面

本人为算法业余爱好者,平时只是随着兴趣偶尔刷刷题,如果上面分享有错误的地方,欢迎指出,感激不尽。