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

【LetMeFly】1764.通过连接另一个数组的子数组得到一个数组

力扣题目链接:https://leetcode.cn/problems/form-array-by-concatenating-subarrays-of-another-array/

给你一个长度为 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 <= 103
  • 1 <= groups[i].length, sum(groups[i].length) <= 103
  • 1 <= nums.length <= 103
  • -107 <= groups[i][j], nums[k] <= 107

方法一:双指针

这道题的数据量是$10^3$级别的,$groups$中所有的数字加起来最多$10^3$个。同时,$nums$数组中的数字个数最多也是$10^3$

因此,$O(sum(len(groups))\times len(nums)$的时间复杂度的算法(最多$10^6$级别)可以通过本题。当然,更高效的KMP算法则更能通过本题

最朴素的算法就是双指针暴力,这道题是要求“按顺序找到所有的groups中的数组”,因此,我们最外层遍历$groups$中的每一个数组,这个数组匹配完了再匹配下一个数组,否则就一直匹配这个数组到$nums$末尾

使用变量$locN$表示$nums$数组“使用”到了哪个元素,接着遍历$groups$中的所有数组,在数组未匹配完时,不断进行匹配,直到这个数组匹配完(接着匹配下一个数组)或nums数组匹配完($nums$中再无元素和匹配,直接返回$false$)或二者不匹配(从$nums$开始匹配这个数组的下一位置开始匹配)为止。

  • 时间复杂度$O(len(nums)\times \max{len(groups[i])})$
  • 空间复杂度$O(1)$

AC代码

C++

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
class Solution {
public:
bool canChoose(vector<vector<int>>& groups, vector<int>& nums) {
int locN = 0;
for (vector<int>& v : groups) {
int locV = 0;
while (locV < v.size() && locN < nums.size()) {
if (v[locV] == nums[locN]) {
locV++, locN++;
}
else {
locN -= locV - 1, locV = 0;
}
}
if (locN == nums.size() && locV != v.size())
return false;
}
return true;
}
};

运行结果还不错:

result

同步发文于CSDN,原创不易,转载请附上原文链接哦~
Tisfy:https://letmefly.blog.csdn.net/article/details/128352023


1764.通过连接另一个数组的子数组得到一个数组
https://blog.letmefly.xyz/2022/12/17/LeetCode 1764.通过连接另一个数组的子数组得到一个数组/
作者
Tisfy
发布于
2022年12月17日
许可协议