使用一个数组map来存储pieces信息,遍历pieces,将每个piece的第一个值作为数组map的index,piece在pieces中的index作为数组map对应的值。(因为 1 <= arr[i], pieces[i][j] <= 100,所以数组map的大小为101即可。并将有值初始化为 -1,方便后面判断)
遍历原数组arr,根据arr[i]从map里找到对应的piece,若找不到(即map[arr[i]] == -1),直接返回false;找到,则遍历piece(从1开始遍历,0位置肯定符合,不用比了),与arr的后续数据比较,有不相等的,直接返回false,否则i++,继续遍历。
若所有遍历完,未返回false,则返回true;
class Solution {
public boolean canFormArray(int[] arr, int[][] pieces) {
int[] map = new int[101];// 因为 1 <= arr[i], pieces[i][j] <= 100
Arrays.fill(map, -1);// 所有值初始化为 -1,方便后面判断
for (int i = 0; i < pieces.length; i++) {
// 每个piece的第一个值作为数组map的index,piece在pieces中的index作为数组map对应的值
map[pieces[i][0]] = i;//记录每个部分的第一个数,方便接上来
}
// 注意:for()这里不要写 i++ 了,也可改用while循环
for (int i = 0; i < arr.length;) {
int pieceIndex = map[arr[i]];
if (pieceIndex != -1) {
int[] piece = pieces[pieceIndex];
i++;
// 从1开始遍历,0位置肯定符合,不用比了
for (int j = 1; j < piece.length; j++) {
if (arr[i] != piece[j]) {
return false;
} else {
i++;
}
}
} else {
return false;
}
}
return true;
}
}