leetcode_力扣_1640. 能否连接形成数组

题目描述

给你一个整数数组 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;
    }
};

提交结果

总结

控制下标。