leetcode-1764-通过连接另一个数组的子数组得到一个数组

给你一个长度为 n 的二维整数数组 groups ,同时给你一个整数数组 nums

你是否可以从 nums 中选出 n不相交 的子数组,使得第 i 个子数组与 groups[i] (下标从 0 开始)完全相同,且如果 i > 0 ,那么第 (i-1) 个子数组在 nums 中出现的位置在第 i 个子数组前面。(也就是说,这些子数组在 nums 中出现的顺序需要与 groups 顺序相同)

如果你可以找出这样的 n 个子数组,请你返回 true ,否则返回 false

如果不存在下标为 k 的元素 nums[k] 属于不止一个子数组,就称这些子数组是 不相交 的。子数组指的是原数组中连续元素组成的一个序列。

示例 1:

输入:groups = [[1,-1,-1],[3,-2,0]], nums = [1,-1,0,1,-1,-1,3,-2,0]
输出:true
解释:你可以分别在 nums 中选出第 0 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 和第 1 个子数组 [1,-1,0,1,-1,-1,3,-2,0] 。
这两个子数组是不相交的,因为它们没有任何共同的元素。

示例 2:

输入:groups = [[10,-2],[1,2,3,4]], nums = [1,2,3,4,10,-2]
输出:false
解释:选择子数组 [1,2,3,4,10,-2] 和 [1,2,3,4,10,-2] 是不正确的,因为它们出现的顺序与 groups 中顺序不同。
[10,-2] 必须出现在 [1,2,3,4] 之前。

示例 3:

输入:groups = [[1,2,3],[3,4]], nums = [7,7,1,2,3,4,7,7]
输出:false
解释:选择子数组 [7,7,1,2,3,4,7,7] 和 [7,7,1,2,3,4,7,7] 是不正确的,因为它们不是不相交子数组。
它们有一个共同的元素 nums[4] (下标从 0 开始)。

提示:

  • groups.length == n
  • 1 <= n <= 10^3
  • 1 <= groups[i].length, sum(groups[i].length) <= 10^3
  • 1 <= nums.length <= 10^3
  • -10^7 <= groups[i][j], nums[k] <= 10^7

分析:

本题本质上就是查找groups[n]是否是nums的子串。由于各个子串不重叠,这个子串越靠前,后续的空间就越大,后面的子串查找到的可能性也越大,因此通过贪心+双指针能够很好地解决这个问题。

代码如下:

class Solution {
public:
    //寻找子串
    bool findSubArray(vector<int>& group, vector<int>& nums, int& begin, int end){
        while(begin < end){
            if(group.size() > (end - begin))//剩余数字不够
                return false;
            int flag = true;
            for(int j = 0;j < group.size();++j){
                if(group[j] != nums[begin + j]){//当前起点取不到子串,begin需要后移
                    flag = false;
                    break;
                }
            }
            if(flag == true){//配对成功
                begin += group.size();
                return true;
            }
            ++begin;
        }
        return false;
    }
    bool canChoose(vector<vector<int>>& groups, vector<int>& nums) {
        int begin = 0;
        int end = nums.size();
        int n = 0;
        while(begin < end){
            if(findSubArray(groups[n],nums,begin,end)){//找到
                if(++n >= groups.size())
                    return true;
            }else{
                return false;
            }
        }
        return false;//此处必须返回false
    }
};

执行用时:12 ms, 在所有 C++ 提交中击败了79.55%的用户

内存消耗:13.1 MB, 在所有 C++ 提交中击败了93.18%的用户

通过测试用例:98 / 98

对于更复杂的测试用例,可能需要KMP算法,这个就下次再写了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值