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