2369. 检查数组是否存在有效划分-dfs深度优先遍历算法

2369. 检查数组是否存在有效划分

给你一个下标从 0 开始的整数数组 nums ，你必须将数组划分为一个或多个 连续 子数组。

如果获得的这些子数组中每个都能满足下述条件 之一 ，则可以称其为数组的一种 有效 划分：

子数组 恰 由 2 个相等元素组成，例如，子数组 [2,2] 。
子数组 恰 由 3 个相等元素组成，例如，子数组 [4,4,4] 。
子数组 恰 由 3 个连续递增元素组成，并且相邻元素之间的差值为 1 。例如，子数组 [3,4,5] ，但是子数组 [1,3,5] 不符合要求。

如果数组 至少 存在一种有效划分，返回 true ，否则，返回 false 。

示例 1：

输入：nums = [4,4,4,5,6]
输出：true
解释：数组可以划分成子数组 [4,4] 和 [4,5,6] 。
这是一种有效划分，所以返回 true 。

示例 2：

输入：nums = [1,1,1,2]
输出：false
解释：该数组不存在有效划分。

这题也挺有意思的，解题代码如下：

int r;
void judge(int *nums,int index,int numsSize,int *rz){
 if(index<numsSize){
     rz[index]=1;
 }
    if(index==numsSize){
        r=1;
    }
    else{
        if(index<=numsSize-3){
            if(nums[index]==nums[index+1]&&nums[index]==nums[index+2]){
                  if(index+3==numsSize){
                    r=1;
                    return ;
                }
                if(r==0&&rz[index+3]==0){
                        judge(nums,index+3,numsSize,rz);

                }
            }
            if(nums[index]+1==nums[index+1]&&nums[index]+2==nums[index+2]){
                if(index+3==numsSize){
                    r=1;
                    return ;
                }
                if(r==0&&rz[index+3]==0){
                        judge(nums,index+3,numsSize,rz);

                }
            }
        }
        if(index<=numsSize-2){
            if(nums[index]==nums[index+1]){
                  if(index+2==numsSize){
                    r=1;
                    return ;
                }
                if(r==0&&rz[index+2]==0){
                        judge(nums,index+2,numsSize,rz);

                }
            }

        }
  }


}

bool validPartition(int* nums, int numsSize){
    r=0;
    int *rz=(int *)malloc(sizeof(int)*numsSize);
    
    for(int i=0;i<numsSize;i++){
        rz[i]=0;
    }
    judge(nums,0,numsSize,rz);
    if(r==1){
        return true;
    }
    else{
        return false;
    }
}