每日力扣-哈希-能否链接形成数组

93 阅读1分钟

给你一个整数数组 arr ,数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces,其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是,不允许 对每个数组 pieces[i] 中的整数重新排序。

如果可以连接 pieces 中的数组形成 arr ,返回 true ;否则,返回 false 。

输入:arr = [49,18,16], pieces = [[16,18,49]] 输出:false 解释:即便数字相符,也不能重新排列 pieces[0]

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

解题思路:题目意思就是arr切成n段,这n段内部顺序不能变。用一个hash存pieces里面每一段的第一个数字映射到第几段。然后第二次遍历每遍历一次arr,就看那一段中的顺序对不对

class Solution {
public:
    bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
        unordered_map<int,int> hash;
        for(int i = 0 ; i < pieces.size() ; i++)
        {
            hash[pieces[i][0]] = i;
        }
        for(int i = 0 ; i < arr.size();)
        {
            if(hash.find(arr[i]) == hash.end()) return false;
            auto &p = pieces[hash[arr[i]]];
            for(int j = 0 ; j < p.size() ; j++,i++)
            {
                if(arr[i] != p[j]) return false;
            }
        }
        return true;
    }
};