【算法训练】LeetCode#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] 属于不止一个子数组，就称这些子数组是 不相交 的。子数组指的是原数组中连续元素组成的一个序列。

示例 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 开始）。

二、思路

题目描述很长，甚至说我都没有什么好的解题思路，不过......

遇到这种问题，最好的解决办法当然是暴力递归了~

三、解题

public class LeetCode1764 {

    public static boolean canChoose(int[][] groups, int[] nums) {
        return process(groups,nums,0,0,groups.length,nums.length);
    }

    public static boolean process(int[][] groups,int[] nums,int gLoc,int nLoc,int gLen,int nLen){
        if (nLoc == -1){
            return false;
        }
        if (nLoc >= nLen){
            // 到达末尾
            return gLoc >= gLen; // / groups也到达末尾则true，否则false
        }
        if (gLoc >= gLen){
            // 只有group结束，nums没结束
            return true;
        }
        for (int i = nLoc ; i < nLen ; ++i){
            if (nums[i] == groups[gLoc][0]){
                // 出现可能相等的子数组
                int nextNLoc = judge(groups[gLoc],nums,i,groups[gLoc].length,nLen);
                if (process(groups,nums,gLoc+1,nextNLoc,gLen,nLen)){
                    // 如果可以
                    return true;
                }
            }
        }
        return false;
    }

    // 判断该子数组是否能够表示这组group，可以则返回下一个位置坐标，不可以则返回-1
    public static int judge(int[] group,int[] nums,int nLoc,int gLen,int nLen){
        for (int i = 0 ; i < gLen ; i++){
            if (i+nLoc > nLen-1){
                // 超过nums界限
                return -1;
            }
            if (group[i] != nums[i+nLoc]){
                return -1;
            }
        }
        return nLoc+gLen;
    }
}