leetcode 3077. K 个不相交子数组的最大能量值【划分型dp+式子等价变形】

2024-03-11 13:44

本文主要是介绍leetcode 3077. K 个不相交子数组的最大能量值【划分型dp+式子等价变形】,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

原题链接:3077. K 个不相交子数组的最大能量值

题目描述:

给你一个长度为 n 下标从 0 开始的整数数组 nums 和一个 正奇数 整数 k 。

x 个子数组的能量值定义为 strength = sum[1] * x - sum[2] * (x - 1) + sum[3] * (x - 2) - sum[4] * (x - 3) + ... + sum[x] * 1 ,其中 sum[i] 是第 i 个子数组的和。更正式的,能量值是满足 1 <= i <= x 的所有 i 对应的 (-1)i+1 * sum[i] * (x - i + 1) 之和。

你需要在 nums 中选择 k 个 不相交子数组 ,使得 能量值最大 。

请你返回可以得到的 最大能量值 。

注意,选出来的所有子数组  需要覆盖整个数组。

输入输出描述:

示例 1:

输入:nums = [1,2,3,-1,2], k = 3
输出:22
解释:选择 3 个子数组的最好方式是选择:nums[0..2] ,nums[3..3] 和 nums[4..4] 。能量值为 (1 + 2 + 3) * 3 - (-1) * 2 + 2 * 1 = 22 。

示例 2:

输入:nums = [12,-2,-2,-2,-2], k = 5
输出:64
解释:唯一一种选 5 个不相交子数组的方案是:nums[0..0] ,nums[1..1] ,nums[2..2] ,nums[3..3] 和 nums[4..4] 。能量值为 12 * 5 - (-2) * 4 + (-2) * 3 - (-2) * 2 + (-2) * 1 = 64 。

示例 3:

输入:nums = [-1,-2,-3], k = 1
输出:-1
解释:选择 1 个子数组的最优方案是:nums[0..0] 。能量值为 -1 。

提示:

  • 1 <= n <= 10^4
  • -10^9 <= nums[i] <= 10^9
  • 1 <= k <= n
  • 1 <= n * k <= 10^6
  • k 是奇数。

解题思路:

本题属于划分类型dp,划分类型dp容易看出来,但是看到这个计算公式有点吓人,这个公式怕是不好处理,首先我们不妨定义f[i][j]表示将前j个数划分出i个子数组的最大能量值,当第j个元素不作为第i组的最后一个元素时,f[i][j]=f[i][j-1],当第j个元素作为第i组的最后一个元素时,对于第i组可以考虑往前延伸,最多延伸到第i个位置,也就是f[i][j]=max(f[i][j],f[i-1][L]+(s[j]-s[L])*w),因为最少要有i-1个元素才有可能划分出i-1组,每一个组至少一个元素,当然j还可以考虑往后延伸,j往后最多延申伸到第n-k+i个位置,因为前面划分了i个子数组,由于总共需要划分k个子数组,所以后面还需要划分k-i个子数组,至少要留下(k-i)个元素才能划分出k-i个子数组,所以我们就可以确定上述表示中L的范围了,(i<=L<=n-k+i),此时我们就可以得到了状态转移方程为f[i][j]=max(f[i][j-1],f[i-1][L]+(s[j]-s[L])*w),i<=L<=n-k+i),但是直接这样做需要枚举划分组数,还要枚举当前位置和前一个延伸的位置,时间复杂度为O(n^2*k),n=1e5,这个复杂度肯定过不了,考虑优化,常见的优化方式就是对式子进行等价变形,或者根据某种性质考虑使用某种数据结构进行优化,我们仔细观察可以发现上述式子可以进行等价变形,上述式子变形为f[i][j]=max(f[i][j-1],s[j]*w+f[i-1][L]-s[L]*w),i<=L<=n-k+i),这个式子中s[j]*w是一个定值,要让值最大就是让f[i-1][L]-s[L]*w最大,其中(i<=L<=n-k+i),可以发现f[i-1][L]-s[L]*w就是前一维,也就是第i-1维的前缀的最大值,我们可以使用一个变量来记录这个前缀最大即可,边枚举j的同时边维护这个前缀最大值,这样就可以优化掉一维,时间复杂度优化为O(n*k),题目说了n*k<=1e6,这个时间肯定是可以过的。

状态定义:

f[i][j]表示前j个元素划分出i个子数组的最大能量值

初始化:

f[0][j]=0,划分0个子数组能量值为0

f[i][i-1]=负无穷,划分i个子数组至少需要i个元素,i-1个元素无法划分出i个子数组,所以初始化为负无穷

状态转移:

f[i][j]=max(f[i][j-1],f[i-1][L]+(s[j]-s[L])*w),i<=L<=n-k+i)

等价转换为:f[i][j]=max(f[i][j-1],s[j]*w+f[i-1][L]-s[L]*w),i<=L<=n-k+i)

最终答案:

最终答案就是将n个元素划分为k个子数组f[k][n]。

时间复杂度:O(n*k)。

空间复杂度:O(n*k)。

cpp代码如下:

typedef long long LL;
const LL mn=-1e18;
class Solution {
public:long long maximumStrength(vector<int>& nums, int k) {int n=nums.size();vector<LL>s(n+1);for(int i=1;i<=n;i++)s[i]=s[i-1]+nums[i-1];vector<vector<LL>>f(k+1,vector<LL>(n+1));for(int i=1;i<=k;i++){f[i][i-1]=mn;  //初始化,至少要i个元素才能划分出i个子数组LL mx=mn;  //记录前缀最大LL w=(k-i+1);if(i%2==0)w*=-1;for(int j=i;j<=n-k+i;j++){mx=max(mx,f[i-1][j-1]-s[j-1]*w);f[i][j]=max(f[i][j-1],s[j]*w+mx);}}return f[k][n];}
};

这篇关于leetcode 3077. K 个不相交子数组的最大能量值【划分型dp+式子等价变形】的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/797962

相关文章

C++原地删除有序数组重复项的N种方法

《C++原地删除有序数组重复项的N种方法》给定一个排序数组,你需要在原地删除重复出现的元素,使得每个元素只出现一次,返回移除后数组的新长度,不要使用额外的数组空间,你必须在原地修改输入数组并在使用O(... 目录一、问题二、问题分析三、算法实现四、问题变体:最多保留两次五、分析和代码实现5.1、问题分析5.

Java中数组转换为列表的两种实现方式(超简单)

《Java中数组转换为列表的两种实现方式(超简单)》本文介绍了在Java中将数组转换为列表的两种常见方法使用Arrays.asList和Java8的StreamAPI,Arrays.asList方法简... 目录1. 使用Java Collections框架(Arrays.asList)1.1 示例代码1.

C++一个数组赋值给另一个数组方式

《C++一个数组赋值给另一个数组方式》文章介绍了三种在C++中将一个数组赋值给另一个数组的方法:使用循环逐个元素赋值、使用标准库函数std::copy或std::memcpy以及使用标准库容器,每种方... 目录C++一个数组赋值给另一个数组循环遍历赋值使用标准库中的函数 std::copy 或 std::

C++初始化数组的几种常见方法(简单易懂)

《C++初始化数组的几种常见方法(简单易懂)》本文介绍了C++中数组的初始化方法,包括一维数组和二维数组的初始化,以及用new动态初始化数组,在C++11及以上版本中,还提供了使用std::array... 目录1、初始化一维数组1.1、使用列表初始化(推荐方式)1.2、初始化部分列表1.3、使用std::

C++ Primer 多维数组的使用

《C++Primer多维数组的使用》本文主要介绍了多维数组在C++语言中的定义、初始化、下标引用以及使用范围for语句处理多维数组的方法,具有一定的参考价值,感兴趣的可以了解一下... 目录多维数组多维数组的初始化多维数组的下标引用使用范围for语句处理多维数组指针和多维数组多维数组严格来说,C++语言没

linux下多个硬盘划分到同一挂载点问题

《linux下多个硬盘划分到同一挂载点问题》在Linux系统中,将多个硬盘划分到同一挂载点需要通过逻辑卷管理(LVM)来实现,首先,需要将物理存储设备(如硬盘分区)创建为物理卷,然后,将这些物理卷组成... 目录linux下多个硬盘划分到同一挂载点需要明确的几个概念硬盘插上默认的是非lvm总结Linux下多

Java 字符数组转字符串的常用方法

《Java字符数组转字符串的常用方法》文章总结了在Java中将字符数组转换为字符串的几种常用方法,包括使用String构造函数、String.valueOf()方法、StringBuilder以及A... 目录1. 使用String构造函数1.1 基本转换方法1.2 注意事项2. 使用String.valu

JAVA中整型数组、字符串数组、整型数和字符串 的创建与转换的方法

《JAVA中整型数组、字符串数组、整型数和字符串的创建与转换的方法》本文介绍了Java中字符串、字符数组和整型数组的创建方法,以及它们之间的转换方法,还详细讲解了字符串中的一些常用方法,如index... 目录一、字符串、字符数组和整型数组的创建1、字符串的创建方法1.1 通过引用字符数组来创建字符串1.2

如何提高Redis服务器的最大打开文件数限制

《如何提高Redis服务器的最大打开文件数限制》文章讨论了如何提高Redis服务器的最大打开文件数限制,以支持高并发服务,本文给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧... 目录如何提高Redis服务器的最大打开文件数限制问题诊断解决步骤1. 修改系统级别的限制2. 为Redis进程特别设置限制

vue如何监听对象或者数组某个属性的变化详解

《vue如何监听对象或者数组某个属性的变化详解》这篇文章主要给大家介绍了关于vue如何监听对象或者数组某个属性的变化,在Vue.js中可以通过watch监听属性变化并动态修改其他属性的值,watch通... 目录前言用watch监听深度监听使用计算属性watch和计算属性的区别在vue 3中使用watchE