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

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

描述

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

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

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

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

示例

输入：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] 。
这两个子数组是不相交的，因为它们没有任何共同的元素。

输入：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 = [[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 开始）。

题解

1. KMP 解法

利用 KMP算法 按顺序进行 groups 子数组的匹配。

时间复杂度：$O(n+m)$，数组的长度总和。
空间复杂度： $O(n)$，用于保存前缀表。

class Solution {
public:
    vector<int> getNext(vector<int> nums){		// 得到子数组的next前缀表
        vector<int> next(nums.size(), 0);
        int k = 0;
        for(int i = 1; i < nums.size(); ++i){
            while(k > 0 && nums[i] != nums[k]){
                k = next[k-1];
            }
            if(nums[i] == nums[k]) ++k;
            next[i] = k;
        }
        return next;
    }
    
    bool canChoose(vector<vector<int>>& groups, vector<int>& nums) {
        int idx = 0, cnt = 0;							// idx为nums数组下标，cnt为匹配成功的数组个数
        int n = groups.size();
        for(int i = 0; i < n; ++i){                     // 匹配第i个数组
            vector<int> next = getNext(groups[i]);  	// next前缀表，用于数组匹配
            int k = 0;
            for( ; idx < nums.size(); ++idx){			// 从idx开始遍历nums
                while(k > 0 && nums[idx] != groups[i][k]){
                    k = next[k-1];						// 更新查找下标，k为group[i]的下标
                }
                if(nums[idx] == groups[i][k]) ++k;		// 元素匹配成功
                if(k == groups[i].size()){				// 子数组匹配成功
                    ++cnt;
                    if(cnt == n) return true;			// 所有数组匹配成功
                    ++idx;								
                    break;								// break 之前要 ++idx
                }
            }
        }
        return false;
    }
};