LeetCode 每日一题——1640. 能否连接形成数组

1.题目描述

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]

2.解题思路与代码

2.1 解题思路

这道题使用哈希表存放 pieces 中每个数字的第一个数字，value 便是该行数组在 pieces 中的索引。然后我们使用 index 指向 arr 数组开始遍历，首先根据 arr 数组的 index 位置上的数字从哈希表中查看是否有该数字的 key ，如果不存在直接返回 false。如果存在则从哈希表中获取出该数组在 pieces 中的索引并将数组提取出来，然后从 index 开始依次与该数组比较，如果存在不同的数字返回 false，匹配相同就重复上面步骤。以 arr = [91,4,64,78], pieces = [[78],[4,64],[91]] 为例，首先我们需要根据 pieces 得到哈希表 map：

然后我们使用 index 从 arr 数组的 0 位置开始，此时 index 指向 91，从 map 中可以看到存在 91 的 key，并且在 pieces 中数组为 [91]，数字相同继续遍历

index 来到 arr 中的数字 4，map 中同样存在数字 4 并且在 pieces 中数组为 [4,64]，依次比较 index=1 是否等于4、index=2 时是否等于 64，比较后相同，继续遍历

此时 index 来到 3 位置数字为 78，map 同样存在 78 并且数组中的值也是 78，遍历完成，返回 true

2.2 代码

class Solution {
    public boolean canFormArray(int[] arr, int[][] pieces) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < pieces.length; i++) {
            map.put(pieces[i][0], i);
        }
        int index = 0;
        while (index < arr.length) {
            if (!map.containsKey(arr[index])) {
                return false;
            }
            int i = map.get(arr[index]);
            for (int j = 0; j < pieces[i].length; j++) {
                if (arr[index] != pieces[i][j]) {
                    return false;
                }
                index++;
            }
        }
        return true;

    }
}

2.3 测试结果

通过测试

3.总结

• 使用哈希表存放 pieces 中的数组第一位和索引
• 根据 arr 遍历时第一位从 map 中取出 pieces 的数组依次比较