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 扁平化成一维数组,数组中的所有整数互不相同)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-array-formation-through-concatenation
2. 思路分析:
分析题目可以知道我们模拟题目描述的过程即可,根据题目可以知道我们需要比对当前的pieces[i]数组中第一个元素开始到pieces[i]结尾的元素是否与arr数组中元素的顺序是一一对应的,假如对应位置不相等那么可以直接返回False了,所以这个时候我们就需要使用map来记录arr中各个元素的位置,并且从题目中可以知道两个数组的元素都是各不相同的,所以使用map是可以记录arr数组中每个元素出现的位置的,因为使用的是python语言所以使用字典来记录arr数组中各个元素出现的位置,我们可以先遍历一遍arr数组记录元素出现的位置,然后遍历pieces数组比对pieces元素在arr中的顺序是否是一样即可
3. 代码如下:
import collections
from typing import List
class Solution:
def canFormArray(self, arr: List[int], pieces: List[List[int]]) -> bool:
dic = collections.defaultdict(int)
for i, n in enumerate(arr):
dic[n] = i
# print(dic)
for i in range(len(pieces)):
l = len(pieces[i])
p = dic[pieces[i][0]]
# 当pieces元素的位置加上pieces[i]的长度大于了arr数组的长度说明无法比对arr与pieces中的全部数字的
if p + l > len(arr): return False
for j in range(l):
# 比对两个数组对应位置的元素
if arr[p] != pieces[i][j]: return False
p += 1
return True