跳转至

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 <= 103
  • 1 <= groups[i].length, sum(groups[i].length) <= 103
  • 1 <= nums.length <= 103
  • -107 <= groups[i][j], nums[k] <= 107

解题思路#

直接转化为字符串匹配。

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
class Solution:
    def canChoose(self, groups: List[List[int]], nums: List[int]) -> bool:
        s = "," + ",".join(map(str, nums)) + ","
        p = 0

        for g in groups:
            t = "," + ",".join(map(str, g)) + ","

            p = s.find(t, p)
            if p == -1:
                return False
            p += len(t) - 1

        return True

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
class Solution {
public:
    bool canChoose(vector<vector<int>>& groups, vector<int>& nums) {
        string s = ",";

        for (auto x : nums) {
            s += to_string(x) + ',';
        }

        int p = 0;

        for (auto &g : groups) {
            string t = ",";

            for (auto x : g) {
                t += to_string(x) + ',';
            }

            p = s.find(t, p);
            if (p == string::npos) {
                return false;
            }
            p += t.size() - 1;
        }

        return true;
    }
};

时间复杂度：\(\mathcal{O}(nums.size()*\max(groups[i].size()))\)
空间复杂度：\(\mathcal{O}(nums.size()+\sum(groups[i].size()))\)

返回顶部

在手机上阅读