Leetcode每日一题 2022-09-22 1640. 能否连接形成数组

最新推荐文章于 2024-07-16 23:15:32 发布

same-pxt

最新推荐文章于 2024-07-16 23:15:32 发布

阅读量47

点赞数

分类专栏： leetcoed每日一题监督学习文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_61509617/article/details/126988975

版权

leetcoed每日一题监督学习专栏收录该内容

17 篇文章 0 订阅

订阅专栏

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;
    }
};