【LeetCode】1640. 能否连接形成数组

题目

1640. 能否连接形成数组

给你一个整数数组 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 扁平化成一维数组，数组中的所有整数互不相同）

思路

• 因为数组中的每个元素各不相同，所以使用哈希表保存每个piece首位对应的piece块
• 遍历数组，根据遍历到的首位查询是否有相等的piece
• 若piece不存在或长度不匹配或值不匹配则直接返回false
• 否则进行下一轮遍历

代码

class Solution:
    def canFormArray(self, arr: List[int], pieces: List[List[int]]) -> bool:
        headMap = {}
        i=0
        for piece in pieces:
            headMap[piece[0]] = piece
        while i<len(arr):
            if arr[i] not in headMap: return False
            j = i + len(headMap[arr[i]])
            if j>len(arr): return False
            if arr[i:j] != headMap[arr[i]]: return False
            i=j
        return True

思路

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$