检查数组有效划分

 

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

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

1. 子数组 恰 由 2 个相等元素组成，例如，子数组 [2,2] 。
2. 子数组 恰 由 3 个相等元素组成，例如，子数组 [4,4,4] 。
3. 子数组 恰 由 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
解释：该数组不存在有效划分。

 思路

动态规划

设数组 nums 的长度为 n，它至少存在一个有效划分的充要条件为：

前 (n−2) 个元素组成的数组至少存在一个有效划分，且后两个元素相等。或

前 (n−3)个元素组成的数组至少存在一个有效划分，且

后三个元素相等。或
后三个元素连续递增，且差值为 1。

int dfs(int i, int n, int* nums)
{

    if (nums[i - 2] == nums[i - 1])
        return 1;
    else
        return 0;
}
int dfs1(int i, int n, int* nums)
{
    if (nums[i - 3] == nums[i - 1] && nums[i - 2] == nums[i - 1])
        return 1;
    if (nums[i - 1] == nums[i - 2] + 1 && nums[i - 2] == nums[i - 3] + 1)
        return 1;
    else
        return 0;
}
bool validPartition(int* nums, int numsSize) {
    int dp[10000000] = { 0 };
    dp[0] = 1;
    for (int i = 1; i <= numsSize; i++)
    {
        if (i >= 2)
        {
            dp[i] = dp[i - 2] && dfs(i, numsSize, nums);
        }
        if (i >= 3)
        {
            dp[i] = dp[i] || (dp[i - 3] && dfs1(i, numsSize, nums));
        }
    }
    return dp[numsSize];

}