算法初探LeetCode-通过连接另一个数组的子数组得到一个数组

开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 30 天，点击查看活动详情

LeetCode1764：通过连接另一个数组的子数组得到一个数组

给你一个长度为 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] 之前。

提示：

• 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数组中的每一个一维数组，依次在nums数组中查找，有任何一个找不到时，返回false，全部找到后返回true

groups数组中的每一个一维数组，依次在nums数组中查找，每次查询有一个查询的起始位置，查找成功后返回下一个数组查找的起始位置。有任何一个找不到时，返回false，全部找到后返回true

算法代码

public boolean canChoose(int[][] groups, int[] nums) {
    int pos = 0;
    for (int i = 0; i < groups.length; i++) {
        pos = canFind(groups[i], nums, pos);
        if (pos == -1) {
            return false;
        }
    }

    return true;
}

private int canFind(int[] group, int[] nums, int pos) {
    if (pos > nums.length - group.length) {
        return -1;
    }
    for (int i = pos; i < nums.length; i++) {
        int j = i;
        while (j < group.length + i && j < nums.length) {
            if (group[j - i] == nums[j]) {
                j++;
            } else {
                break;
            }
        }
        if (j == group.length + i) {
            return j;
        }
    }
    return -1;
}

结果详情

算法复杂度

• 空间复杂度：$O(n)$
• 时间复杂度：$O(n)$

在掘金(JUEJIN)一起进步，一起成长！