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

Raphael Liu Lv10
  2023-09-13 00:47:26 2023-09-13 00:47:26 Created   2023-09-13 16:06:23 2023-09-13 16:06:23 Updated    

给你一个下标从 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
**解释：** 该数组不存在有效划分。

提示：

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

直接定义状态：dp[i+1]表示[0,n]范围内是否可以被成功划分，这样只用对3种情况分别讨论即可。
若满足三种条件之一，且dp[i+1-k](k = 2,3)也为true，则当前点也可以被划分，dp[i+1] = true。

[]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
class Solution {
    public boolean validPartition(int[] nums) {
        int n = nums.length;
        boolean[] dp = new boolean[n+1];//dp[i+1]表示[0,i]是否可以被成功划分
        dp[0] = true;
        for (int i = 1; i < n; i++) {
            if (nums[i] == nums[i-1] && dp[i-1]) dp[i+1] = true;
            if (i > 1) {
                if (nums[i] == nums[i-1] && nums[i] == nums[i-2] && dp[i-2]) dp[i+1] = true;
                if (nums[i] == nums[i-1]+1 && nums[i] == nums[i-2]+2 && dp[i-2]) dp[i+1] = true;
            }
        }
        return dp[n];
    }
}
[]
1
2
3
4
5
6
7
8
9
10
11
12
13
14
class Solution:
    def validPartition(self, nums: List[int]) -> bool:
        n = len(nums)
        dp = [False for _ in range(n+1)] # dp[i+1]表示[0,i]是否可以被成功划分
        dp[0] = True
        for i in range(1,n):
            if nums[i] == nums[i-1] and dp[i-1]:
                dp[i+1] = True
            if i > 1:
                if nums[i] == nums[i-1] == nums[i-2] and dp[i-2]:
                    dp[i+1] = True
                if nums[i] == nums[i-1]+1 == nums[i-2]+2 and dp[i-2]:
                    dp[i+1] = True
        return dp[n]
 Comments
On this page
2369-检查数组是否存在有效划分