图解LeetCode——1640. 能否连接形成数组（难度：简单）

一、题目

给你一个整数数组 arr ，数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces，其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是，不允许 对每个数组 pieces[i] 中的整数重新排序。

如果可以连接 pieces 中的数组形成 arr ，返回 true ；否则，返回 false 。

二、示例

2.1> 示例 1：

【输入】arr = [15,88], pieces = [[88],[15]] 【输出】true 【解释】依次连接 [15] 和 [88]

2.2> 示例 2：

【输入】arr = [49,18,16], pieces = [[16,18,49]] 【输出】false 【解释】即便数字相符，也不能重新排列 pieces[0]

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

三、解题思路

根据题意，我们首先创建一个数组index，用于存储pieces中每个数组元素的下标与该数组中第一个元素值的对应关系，即：index[pieces[i][0]] = i;

然后，我们遍历数组arr，拿着遍历到的元素去index数组中查找，找到对应的下标i，那么这个下标i就是该元素所在的数组在pieces中的下标位置了。

同时遍历arr和pieces中的元素，只要拿某个元素发现在index中存储的值为-1，或者pieces中数组元素内部排序不对，则直接返回false，不需要继续执行下去。否则，全部执行完毕后，返回true。详情如下图所示：

四、代码实现

class Solution {
    public boolean canFormArray(int[] arr, int[][] pieces) {
        int[] index = new int[101];
        Arrays.fill(index, -1);
        for (int i = 0; i < pieces.length; i++) index[pieces[i][0]] = i;     
        for (int i = 0; i < arr.length;) {
            if (index[arr[i]] == -1) return false;
            for (int item : pieces[index[arr[i]]]) {
                if (arr[i] != item) return false;
                i++;
            }
        }
        return true;
    }
}