[LeetCode 1640]能否连接形成数组

题目描述

题目链接：[LeetCode 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 扁平化成一维数组，数组中的所有整数互不相同）

思路分析

数量级不大，直接暴力枚举每个数组的开头即可

代码

class Solution {
public:
    bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
        for(int i = 0; i < arr.size(); i++)
        {
            //判断一轮下来，有没有找到开头与当前元素相同的数组
            bool jg = false;
            for(int j = 0; j < pieces.size(); j++)
            {
                if(arr[i] == pieces[j][0])
                {
                    jg = true;
                    int k = 0;
                    while(k < pieces[j].size() && pieces[j][k] == arr[i]) i++, k++;
                    if(k < pieces[j].size()) return false;

                    break;
                }
            }  
            //如果没找到，比如[49,18,16], pieces = [[16,18,49]]，直接返回false，不然会陷入循环           
            if(!jg) return false;
            i--;
        }

        return true;
    }
};