【动态规划-划分型 DP】力扣2369. 检查数组是否存在有效划分

news/2024/11/8 23:38:20 标签: 动态规划, leetcode, 算法

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

在这里插入图片描述

动态规划

class Solution {
public:
    bool validPartition(vector<int>& nums) {
        int n = nums.size();
        vector<int> dp(n+1, 0);
        if(n == 2){
            return nums[1] == nums[0];
        }
        dp[0] = 1;
        dp[2] = nums[1] == nums[0];
        for(int i = 3; i < n+1; i++){
            if(nums[i-1] == nums[i-2]){
                dp[i] |= dp[i-2];
            }
            if(nums[i-1] == nums[i-2] && nums[i-2] == nums[i-3]){
                dp[i] |= dp[i-3];
            }
            if(nums[i-1] == nums[i-2]+1 && nums[i-2] == nums[i-3]+1){
                dp[i] |= dp[i-3];
            }
        }
        return dp[n];
    }
};

时间复杂度:O(n),其中 n 是数组 nums 的长度。我们仅遍历一次 nums。
时间复杂度:O(n)。数组 dp 消耗 O(n)。

我们定义一个dp,用来储存在前i个元素中,是否有至少一个有效划分。我们在遍历过程中,一旦发现此时最近的几个元素,能够满足三种有效划分情况的一种,我们根据在这几个元素之前是否也存在有效划分,如果存在,那么说明目前至少存在一种有效划分。

最后返回dp[n]即可。

空间优化

class Solution {
public:
    bool validPartition(vector<int>& nums) {
        int n = nums.size();
        if (n == 2) {
            return nums[1] == nums[0];
        }

        // 只需要三个变量来存储之前的状态
        bool dp0 = true;              // 对应于 dp[i-3]
        bool dp1 = false;             // 对应于 dp[i-2]
        bool dp2 = nums[1] == nums[0]; // 对应于 dp[i-1]
        bool current = false;          // 用于存储 dp[i] 的值

        for (int i = 3; i <= n; i++) {
            current = false;
            if (nums[i - 1] == nums[i - 2]) {
                current |= dp1;
            }
            if (nums[i - 1] == nums[i - 2] && nums[i - 2] == nums[i - 3]) {
                current |= dp0;
            }
            if (nums[i - 1] == nums[i - 2] + 1 && nums[i - 2] == nums[i - 3] + 1) {
                current |= dp0;
            }
            
            // 更新 dp0, dp1, dp2
            dp0 = dp1;
            dp1 = dp2;
            dp2 = current;
        }

        return current;
    }
};

空间压缩至O(1)

通过对未压缩代码观察,我们发现我们在动态规划状态转移过程中,最多只用到当前元素和前面两个元素,那么说明我们可以用三个变量来储存这些状态,然后在遍历nums的过程中不断覆盖滚动。


http://www.niftyadmin.cn/n/5744562.html

相关文章

[产品管理-58]:安索夫矩阵矩阵帮助创业者确定研发出来的产品在市场中定位策略

目录 一、提出背景 二、核心思想与结构 三、应用背景与领域 四、实践案例 安索夫矩阵&#xff08;Ansoff Matrix&#xff09;&#xff0c;也被称为产品/市场方格或成长矢量矩阵&#xff0c;其应用背景可以从以下几个方面进行详细阐述&#xff1a; 一、提出背景 安索夫矩阵…

大数据Informatica面试题及参考答案

目录 什么是 Informatica?它主要解决什么问题? 什么是 Informatica PowerCenter? Informatica PowerCenter 的主要组成部分有哪些? 解释 Informatica PowerCenter 的主要组件。 Informatica PowerCenter 与 DataStage 有何区别? 解释 Informatica 中的源 (Source) 和…

Python邮差:如何用代码精确投递商品快递费用的密信

目录 一、准备工作 二、编写API请求脚本 三、解析与处理快递费用数据 四、案例应用&#xff1a;模拟电商平台的快递费用计算 五、自动化邮件通知 六、总结 在电子商务的广阔天地里&#xff0c;精确计算并快速传递商品快递费用是一项至关重要的任务。作为Python邮差&#…

高速电机的设计有七个主要问题你知道吗?

在电机技术不断进步的今天&#xff0c;高速电机因其在诸多应用场景中展现出的高效能和紧凑性而备受瞩目。然而&#xff0c;设计一款性能优良的高速电机是一项复杂而富有挑战性的任务。 一、热管理问题 在高速运行条件下&#xff0c;高速电机会产生大量热量&#xff0c;这会直…

SpringBoot整合SpringSecurity实现密码加密解密、登录认证退出功能

文章目录 SpringBoot整合SpringSecurity实现密码加密解密、登录认证退出功能一、引言二、JWT简介与组成1、JWT简介2、JWT的组成2.1、Header&#xff08;头部&#xff09;2.2、Payload&#xff08;载荷&#xff09;2.3、Signature&#xff08;签名&#xff09; 三、Spring Secur…

Java LeetCode练习

3194. 最小元素和最大元素的最小平均值 package JavaExercise;import java.util.Arrays;public class Exercise {public static void main(String[] args) {int[] array {1,2,3,7,8,9};System.out.println(Solution.minimumAverage(array));} }class Solution {public static…

70B的模型做微调,使用A10*8的卡能够使用

使用 8 张 A10 GPU&#xff08;每张 A10 GPU 大约有 24 GB 的显存&#xff09;来微调 70B 参数的模型会比较困难&#xff0c;主要原因是显存不足。像 70B 参数量级的模型&#xff08;如 LLaMA-2 70B、BLOOM-176B&#xff09;通常需要几百 GB 以上的显存&#xff0c;仅加载模型就…

计算机专业开题报告写法,该怎么写好?

不会写开题报告&#xff0c;或者想要一些论文模版的&#xff0c;欢迎评论&#xff0c;会第一时间给大家。 题报告是计算机专业大学毕业生在开展毕业设计或论文研究前&#xff0c;对研究课题进行详细介绍和计划的重要环节。作为开题者对科研课题的一种文字说明&#xff0c;开题…