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

相关文章

哈希leetcode-1

目录 1前言 2.例题  2.1两数之和 2.2判断是否互为字符重排 2.3存在重复元素1 2.4存在重复元素2 2.5字母异位词分组 1前言 哈希表主要是适合于快速查找某个元素(O(1)) 当我们要频繁的查找某个元素,第一哈希表O(1),第二,二分O(log n) 一般可以分为语言自带的容器哈希和用数组模拟的简易哈希。 最简单的比如数组模拟字符存储,只要开26个c

csu(背包的变形题)

题目链接 这是一道背包的变形题目。好题呀 题意:给n个怪物,m个人,每个人的魔法消耗和魔法伤害不同,求打死所有怪物所需的魔法 #include<iostream>#include<algorithm>#include<cstring>#include<stack>#include<queue>#include<set>//#include<u>#include<map

hdu2241(二分+合并数组)

题意:判断是否存在a+b+c = x,a,b,c分别属于集合A,B,C 如果用暴力会超时,所以这里用到了数组合并,将b,c数组合并成d,d数组存的是b,c数组元素的和,然后对d数组进行二分就可以了 代码如下(附注释): #include<iostream>#include<algorithm>#include<cstring>#include<stack>#include<que

hdu4826(三维DP)

这是一个百度之星的资格赛第四题 题目链接:http://acm.hdu.edu.cn/contests/contest_showproblem.php?pid=1004&cid=500 题意:从左上角的点到右上角的点,每个点只能走一遍,走的方向有三个:向上,向下,向右,求最大值。 咋一看像搜索题,先暴搜,TLE,然后剪枝,还是TLE.然后我就改方法,用DP来做,这题和普通dp相比,多个个向上

hdu1011(背包树形DP)

没有完全理解这题, m个人,攻打一个map,map的入口是1,在攻打某个结点之前要先攻打其他一个结点 dp[i][j]表示m个人攻打以第i个结点为根节点的子树得到的最优解 状态转移dp[i][ j ] = max(dp[i][j], dp[i][k]+dp[t][j-k]),其中t是i结点的子节点 代码如下: #include<iostream>#include<algorithm

hdu3389(阶梯博弈变形)

题意:有n个盒子,编号1----n,每个盒子内有一些小球(可以为空),选择一个盒子A,将A中的若干个球移到B中,满足条件B  < A;(A+B)%2=1;(A+B)%3=0 这是阶梯博弈的变形。 先介绍下阶梯博弈: 在一个阶梯有若干层,每层上放着一些小球,两名选手轮流选择一层上的若干(不能为0)小球从上往下移动,最后一次移动的胜出(最终状态小球都在地面上) 如上图所示,小球数目依次为

hdu4865(概率DP)

题意:已知前一天和今天的天气概率,某天的天气概率和叶子的潮湿程度的概率,n天叶子的湿度,求n天最有可能的天气情况。 思路:概率DP,dp[i][j]表示第i天天气为j的概率,状态转移如下:dp[i][j] = max(dp[i][j, dp[i-1][k]*table2[k][j]*table1[j][col] )  代码如下: #include <stdio.h>#include

usaco 1.1 Broken Necklace(DP)

直接上代码 接触的第一道dp ps.大概的思路就是 先从左往右用一个数组在每个点记下蓝或黑的个数 再从右到左算一遍 最后取出最大的即可 核心语句在于: 如果 str[i] = 'r'  ,   rl[i]=rl[i-1]+1, bl[i]=0 如果 str[i] = 'b' ,  bl[i]=bl[i-1]+1, rl[i]=0 如果 str[i] = 'w',  bl[i]=b

hdu 1166 敌兵布阵(树状数组 or 线段树)

题意是求一个线段的和,在线段上可以进行加减的修改。 树状数组的模板题。 代码: #include <stdio.h>#include <string.h>const int maxn = 50000 + 1;int c[maxn];int n;int lowbit(int x){return x & -x;}void add(int x, int num){while

uva 10154 DP 叠乌龟

题意: 给你几只乌龟,每只乌龟有自身的重量和力量。 每只乌龟的力量可以承受自身体重和在其上的几只乌龟的体重和内。 问最多能叠放几只乌龟。 解析: 先将乌龟按力量从小到大排列。 然后dp的时候从前往后叠,状态转移方程: dp[i][j] = dp[i - 1][j];if (dp[i - 1][j - 1] != inf && dp[i - 1][j - 1] <= t[i]