• 欢迎光临~

1640. 能否连接形成数组

开发技术 开发技术 2021-04-06 130次浏览

题目:给你一个整数数组 arr ,数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces,其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是,不允许 对每个数组 pieces[i] 中的整数重新排序。如果可以连接 pieces 中的数组形成 arr ,返回 true ;否则,返回 false 。

 示例 1:

输入:arr = [85], pieces = [[85]]
输出:true
示例 2:

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

示例 3:

输入:arr = [91,4,64,78], pieces = [[78],[4,64],[91]]
输出:true
解释:依次连接 [91]、[4,64] 和 [78]

题解:

class Solution {
public:
    bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
        unordered_map<int, int> dic;
        for (int i = 0; i < pieces.size(); i++) {
            dic[pieces[i][0]] = i;
        }

        for (int i = 0; i < arr.size();) {
            if (dic.find(arr[i]) == dic.end()) return false;
            auto& p = pieces[dic[arr[i]]];
            for (int j = 0; j < p.size(); j++, i++) {
                if (arr[i] != p[j]) return false;
            }
        }
        return true;
    }
};
程序员灯塔
转载请注明原文链接:1640. 能否连接形成数组
喜欢 (0)