力扣 (LeetCode) - 排序-刷题-面试题 1640-能否连接形成数组

目录

题目描述：

思路分析：

我的答案：

---

题目描述：

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

---

思路分析：

参照题目下方的提示：

提示

意思是：pieces数组中的每一个元素都能匹配到arr数组中的一段。那么就可以

1）先判断pieces数组中每一个元素的第一个数字是否存在于arr中

2）如果不存在就返回false，如果存在，就把这个元素匹配到arr中，匹配不到就返回false

---

我的答案：

class Solution(object):
    def canFormArray(self, arr, pieces):
        """
        :type arr: List[int]
        :type pieces: List[List[int]]
        :rtype: bool
        """
        
        flag=True
        for i in pieces:
            if (i[0] not in arr) or (arr[arr.index(i[0]):arr.index(i[0])+len(i)]!=i):
                flag=False
        return flag