leetcode_1640 能否连接形成数组

70 阅读1分钟

要求

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

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

示例 1:

输入:arr = [85], pieces = [[85]]
输出:true

示例 2:

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

示例 3:

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

示例 4:

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

示例 5:

输入:arr = [1,3,5,7], pieces = [[2,4,6,8]]
输出:false

提示:

  • 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 扁平化成一维数组,数组中的所有整数互不相同)

核心代码

class Solution:
    def canFormArray(self, arr: List[int], pieces: List[List[int]]) -> bool:
        lenth = len(arr)
        i = 0
        while i <= lenth - 1:
            is_find = False
            for each in pieces:
                if arr[i] in each:
                    if arr[i:i + len(each)] != each:
                        return False
                    else:
                        i += len(each)
                        is_find = True
                        break
            else:
                if not is_find:
                    return False
        return True

image.png

解题思路:用i计数,用is_find标记是否找到,遍历列表arr,然后在列表pieces中寻找该元素, 如果找到,就对列表arr用切片切出与pieces中小列表长度相同的部分,对两个长度相同的列表进行对比,如果不相同则返回False,如果相同则将i加上小列表的长度,将is_find设为True,然后跳出循环;如果循环结束且没有找到(is_find还是False),就返回False; 如果都能形成连接,则返回True,比较简单。