leetcode 1640.能否连接形成数组

1640. 能否连接形成数组

难度
简单

132

 

 

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

 

思路:

哈希表标记每个整数数组头部元素，遍历arr数组，每找到一个碎片数组的头部，比较其后面的元素和arr是否相同，如果全部相同，改变遍历下标寻找下一个碎片数组的头部，如果没找到该头部，直接返回false 如果全部比对通过，返回true 

class Solution {
public:
    bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
         unordered_map<int,int> um;
         for(int i=0;i<pieces.size();i++){
             um[pieces[i][0]]=i;
         }
         int index=0,n=arr.size();
         while(index<n){
             if(um.find(arr[index])!=um.end()) {
                 vector<int>&p=pieces[um[arr[index]]];
                 for(int i=0;i<p.size();i++){
                     if(arr[index+i]!=p[i]) return false;
                 }
                 index+=p.size();
             }
            else return false;
         }
         return true;
    }
};