Leetcode每日一题 2022-09-22 1640. 能否连接形成数组
水题一道,用哈希表记录pieces里的首元素,然后遍历arr,如果没法找到首元素直接false,找到后向后遍历,看是否一直相等。
class Solution {
public:
bool canFormArray(vector<int> &arr, vector<vector<int>> &pieces) {
unordered_map<int, int> index;
for (int i = 0; i < pieces.size(); i++) {
index[pieces[i][0]] = i;
}
for (int i = 0; i < arr.size();) {
auto it = index.find(arr[i]);
if (it == index.end()) {
return false;
}
for (int x : pieces[it->second]) {
if (arr[i++] != x) {
return false;
}
}
}
return true;
}
};