力扣leetcode 1640. 能否连接形成数组

CSDN话题挑战赛第2期
参赛话题：算法题解

题目链接与描述

https://leetcode.cn/problems/check-array-formation-through-concatenation/

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

关键词：

方法一：hash 表

运行截图

代码

  public boolean canFormArray(int[] arr, int[][] pieces) {
        // 不能重新排序，所以挨个尝试比较+回溯即可得到是否能组成arr
        // 循环arr每个元素，记录元素下标
        // 循环对比pieces ，再记录arr对比当前下标，并移动pieces 下标，直到数组pieces全部使用完成
        // 如果满足则下移pieces ，arr当前下标 结束 返回true
        // 全部满足则使用数组累加1
        // 如果不满足则继续下一个pieces中的数组，恢复循环中的arr下标
        // 这种方式需要很多次循环，并且可能需要在pieces中反复扫，所以拿到提示中的：
        // arr 中的整数 互不相同
        // pieces 中的整数 互不相同
        // 1 <= arr[i], pieces[i][j] <= 100
        // 我们可以构建一个pieces 索引，找到对应的下标，因为预存可能需要100的空数组，很多空间浪费了所以用map
        Map<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < pieces.length; i++) {
            map.put(pieces[i][0], i);
        }
        for (int i = 0; i < arr.length; ) {
            Integer index = map.get(arr[i]);
            if (index == null) {
                return false;
            }
            for (int item : pieces[index]) {
                if (item != arr[i++]) {
                    return false;
                }
            }
        }
        return true;
    }

结尾

欢迎评论区交流，每日打卡，冲冲冲！！！