1640. Check Array Formation Through Concatenation

LeetCode
, ,

今天的题目是：1640. 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]

简单题，用一个hash表记录住每个 pieces 第一个元素到pieces的索引即可，然后模拟跑一边即可。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
	if (arr.size() == 0) return true;
	unordered_map<int, int> imap; // pieces[index][0] -> index
	for(int i = 0;i < pieces.size(); i++) {
		imap[pieces[i][0]] = i;
	}

	for(int i = 0;i < arr.size();) {
		if (imap.count(arr[i]) == 0) return false;
		auto& piece = pieces[imap[arr[i]]];
		for (auto v : piece) {
			if (i >= arr.size() || v != arr[i]) return false;
			i++;
		}
	}
	return true;
}