leetcode笔记之[1640. 能否连接形成数组 ]

100 阅读1分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第14天,点击查看活动详情

一、题目描述:

1640. 能否连接形成数组 - 力扣(LeetCode)

给你一个整数数组 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 扁平化成一维数组,数组中的所有整数互不相同)

二、思路分析:

从遍历 pieces 数组入手,再遍历其中的小数组, 在 arr 中寻找小数组中每个元素的位置

如果没找到,自然不用再继续,返回 false 即可。

如果找到了就记录此元素出现的位置,再寻找下一个,如果没找到,返回 false

如果找到,检查此元素在 arr 中的位置与上一个元素在 arr 中 的位置是否相差 1

差 1 则继续,否则返回 false。当 pieces 被顺利遍历完,说明没有失序的数字

最后返回 true 即可。

得益于数组中元素不会重复,如果有重复元素,那这个方法可能不太得行,还需改进 :)

三、AC 代码:

class Solution {
public:
    bool canFormArray(vector<int>& arr, vector<vector<int>>& pieces) {
        for (auto p : pieces) { //遍历 pieces
            auto np = p.size(); //数组中元素个数
            int index(0);
            for (size_t i = 0; i < np; ++i) { //遍历 p 中每个元素
                auto pos = find(arr.begin(), arr.end(), p[i]); //在 arr 中寻找是否存在该元素
                if (pos == arr.end()) { // arr 中找不到
                    return false;
                }else{
                    auto tmp = pos-arr.begin(); //获得此元素在 arr 中的位置
                    if (i == 0){ //第一个元素只需记录其在 arr 中的位置
                        index = tmp;
                    }else{
                        if (tmp-index != 1){ //除了第 1 个元素,在找到位置的同时还要判断
                                             //与前一个元素在 arr 中的位置是否相差1
                            return false;
                        }
                        index = tmp; //更新位置
                    }
                }
            }
        }
        return true;
    }
};