leetcode刷题记录-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
解释：该数组不存在有效划分。

提示：

2 <= nums.length <= 105
1 <= nums[i] <= 106

---

思路

动态规划。
如果数组存在有效划分，则它的前两个数或者前三个数满足题目条件。划去这个划分数组后，剩下的数组里同样的前两个或三个数满足条件，以此类推。
因此最后要解决的问题就是数组的前i个数能否有效划分。
设置boolean类型dp数组，dp[n]表示长度为n的数组是否存在有效划分。每次遍历判断前两个数和前三个数是否符合条件。

实现

class Solution {
    public boolean validPartition(int[] nums) {
        int n = nums.length;
        if (n == 2) {
            return nums[0] == nums[1];
        }
        boolean[] dp = new boolean[n+1];
        dp[0] =true;
        for (int i = 1; i <= n; i++) {
            if (i >= 2) {
                dp[i] = nums[i - 2] == nums[i - 1] && dp[i - 2];
            }
            if (i >= 3) {
                dp[i] = dp[i]
                        || (dp[i - 3]
                        && ((nums[i - 3] == nums[i - 2] && nums[i - 2] == nums[i - 1])
                        || ((nums[i - 3] + 1) == nums[i - 2] && nums[i - 2] == (nums[i - 1] - 1))));
            }
        }
        return dp[n];
    }
}

---

总结

动态规划入门题，关键是找出初始状态，状态转移方程。