题目描述
给你一个整数数组 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 {
public:
int get_pos(const vector< vector<int> > pieces, int key){ // 在 pieces 中寻找可能符合要求的数组的位置
int i = 0;
int pieces_len = pieces.size();
for(i = 0; i < pieces_len; i++){
if(pieces[i][0] == key){
return i;
}
}
return -1;
}
int updata_idx(vector<int> arr, int old_idx, vector<int> pieces_sub){ // 判断对应位置下的数组是否可以匹配,若匹配更新 arr 数组的 idx,不匹配返回原先的 idx
int arr_len = arr.size();
int sub_len = pieces_sub.size();
int i, j;
i = old_idx;
j = 0;
while(i < arr_len && j < sub_len){
if(arr[i] == pieces_sub[j]){
i++;
j++;
continue;
}
break;
}
if(j == sub_len){
return i;
}
return old_idx;
}
bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
int arr_len = arr.size();
int idx, pos;
idx = 0;
while(1){
if(idx >= arr_len){
break;
}
pos = get_pos(pieces, arr[idx]);
if(pos == -1){
return false;
}
idx = updata_idx(arr, idx, pieces[pos]);
pieces[pos][0] = -1; // 为匹配的数组添加标记
}
return true;
}
};
提交结果
总结
控制下标。