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 {
/**
* 将 arr 中所有数字的下标放入哈希表方便查询
* 遍历 pieces 中每一个数组时, 查询 pieces[i][0] 是否在 arr 中存在,
* 如果不存在, 返回false, 如果存在对比 arr 跟 pieces[i] 顺序
*/
public boolean canFormArray(int[] arr, int[][] pieces) {
Map<Integer, Integer> map = new HashMap<>();
int len = arr.length;
for (int i = 0; i < len; i++) {
map.put(arr[i], i);
}
for (int[] piece : pieces) {
int index = map.getOrDefault(piece[0], len);
for (int i = 0; i < piece.length; i++, index++) {
if (index == len || arr[index] != piece[i]) {
return false;
}
}
}
return true;
}
}