Keshawn_lu's Blog

Leetcode 1640. 能否连接形成数组

字数统计: 431阅读时长: 1 min
2022/09/22 Share

题目简介:

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

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

示例 1:

1
2
3
输入:arr = [15,88], pieces = [[88],[15]]
输出:true
解释:依次连接 [15] 和 [88]

示例 2:

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

提示:

  • 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 扁平化成一维数组,数组中的所有整数互不相同)

思路:

利用哈希表,将pieces中每个数组的第一个元素作为key,数组作为value进行存储。

然后对arr进行遍历,若arr中的元素没有在mapkey中,则直接返回false;遇到key后,则观察arr后续元素是否和value中元素一一对应即可。

代码如下:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
class Solution {
public:
bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {

unordered_map<int, vector<int>> map;

for(int i = 0; i < pieces.size(); i++){

map[pieces[i][0]]= pieces[i];
}

for(int i = 0; i < arr.size(); i++){

if(map.count(arr[i]) == 0)
return false;

int k = i;
for(int j = 1; j < map[arr[i]].size(); j++){

k++;
if(map[arr[i]][j] != arr[k])
return false;
}

i = k;
}

return true;
}
};
CATALOG
  1. 1. 题目简介:
  2. 2. 思路:
  3. 3. 代码如下: