🙉饭不食,水不饮,题必须刷🙉
C语言免费动漫教程,和我一起打卡! 🌞《光天化日学C语言》🌞
LeetCode 太难?先看简单题! 🧡《C语言入门100例》🧡
数据结构难?不存在的! 🌳《画解数据结构》🌳
LeetCode 太简单?算法学起来! 🌌《夜深人静写算法》🌌
一、题目
1、题目描述
给定一个整数数组
nums
和一个整数 k k k ,请返回 非空 子序列元素和的最大值,子序列需要满足:子序列中每两个 相邻 的整数nums[i]
和nums[j]
,它们在原数组中的下标 i i i 和 j j j 满足 i < j i < j i<j 且 j − i ≤ k j - i \le k j−i≤k。
数组的子序列定义为:将数组中的若干个数字删除(可以删除 0 个数字),剩下的数字按照原本的顺序排布。
样例输入:nums = [10,2,-10,5,20], k = 2
样例输出:37
2、基础框架
- C语言版本 给出的基础框架代码如下:
int constrainedSubsetSum(int* nums, int numsSize, int k){
}
3、原题链接
( 1 ) (1) (1) LeetCode 1425. 带限制的子序列和
二、解题报告
1、思路分析
首先,容易想到状态转移方程
d
p
[
i
]
=
n
u
m
s
[
i
]
+
max
j
=
i
−
k
i
−
1
d
p
[
j
]
dp[i] = nums[i] + \max_{j=i-k}^{i-1} dp[j]
dp[i]=nums[i]+j=i−kmaxi−1dp[j] 这是一个时间复杂度为
O
(
n
k
)
O(nk)
O(nk) 的算法;
我们考虑
max
j
=
i
−
k
i
−
1
d
p
[
j
]
\max_{j=i-k}^{i-1} dp[j]
maxj=i−ki−1dp[j] 这一段,如果能够通过
O
(
1
)
O(1)
O(1) 的时间复杂度来获取,这样总的时间复杂度就变成了
O
(
n
)
O(n)
O(n),这个式子展开后得到
m
a
x
(
d
p
[
i
−
k
]
,
d
p
[
i
−
k
+
1
]
,
.
.
.
,
d
p
[
i
−
1
]
)
max( dp[i-k], dp[i-k+1], ..., dp[i-1] )
max(dp[i−k],dp[i−k+1],...,dp[i−1]) ,于是,我们可以维护一个
(
d
p
[
i
−
k
]
,
d
p
[
i
−
k
+
1
]
,
.
.
.
,
d
p
[
i
−
1
]
)
( dp[i-k], dp[i-k+1], ..., dp[i-1] )
(dp[i−k],dp[i−k+1],...,dp[i−1]) 的队列;
问题转化成求这个队列中的最大值,由于队列只支持取队列头操作,所以队列头的必然是最大值,即这是一个单调递减的队列。加入队列中的两个元素
i
<
j
i < j
i<j 且
d
p
[
i
]
≤
d
p
[
j
]
dp[i] \le dp[j]
dp[i]≤dp[j],那么
d
p
[
i
]
dp[i]
dp[i] 不会比
d
p
[
j
]
dp[j]
dp[j] 更优,直接舍弃即可,所以整个队列中的元素一定满足
i
<
j
i < j
i<j 时,
d
p
[
i
]
>
d
p
[
j
]
dp[i] > dp[j]
dp[i]>dp[j],即它是一个 单调递减队列;
迭代计算
d
p
[
i
]
dp[i]
dp[i] 的值即可。 取
max
i
=
0
n
−
1
d
p
[
i
]
\max_{i=0}^{n-1} dp[i]
maxi=0n−1dp[i] 就是最后的答案了。
2、时间复杂度
- 任何一个元素只会 入队 和 出队 各一次,总的时间复杂度为 O ( n ) O(n) O(n)。均摊下来每次的操作就是 O ( 1 ) O(1) O(1) 的。
3、代码详解
/**************************** 顺序表 实现队列 ****************************/
#define DataType int
#define maxn 100005
struct Queue {
DataType data[maxn];
int head, tail;
};
void QueueClear(struct Queue* que) {
que->head = que->tail = 0;
}
void QueueEnqueue(struct Queue *que, DataType dt) {
que->data[ que->tail++ ] = dt;
}
void QueueDequeueFront(struct Queue* que) {
++que->head;
}
void QueueDequeueRear(struct Queue* que) {
--que->tail;
}
DataType QueueGetFront(struct Queue* que) {
return que->data[ que->head ];
}
DataType QueueGetRear(struct Queue* que) {
return que->data[ que->tail - 1 ];
}
int QueueGetSize(struct Queue* que) {
return que->tail - que->head;
}
int QueueIsEmpty(struct Queue* que) {
return !QueueGetSize(que);
}
/**************************** 顺序表 实现队列 ****************************/
int dp[maxn];
int constrainedSubsetSum(int* nums, int numsSize, int k){
struct Queue *q = (struct Queue *)malloc( sizeof(struct Queue) );
int i, max;
max = dp[0] = nums[0]; // (1)
QueueClear(q);
QueueEnqueue(q, 0); // (2)
for(i = 1; i < numsSize; ++i) { // (3)
while(!QueueIsEmpty(q) && i - QueueGetFront(q) > k) // (4)
QueueDequeueFront(q);
dp[i] = nums[i] + dp[ QueueGetFront(q) ]; // (5)
if (dp[ QueueGetFront(q) ] < 0)
dp[i] = nums[i]; // (6)
while(!QueueIsEmpty(q) && dp[ QueueGetRear(q) ] <= dp[i]) // (7)
QueueDequeueRear(q);
QueueEnqueue(q, i); // (8)
if(dp[i] > max)
max = dp[i]; // (9)
}
return max;
}
- ( 1 ) (1) (1) 初始化第 0 个元素的值;
- ( 2 ) (2) (2) 将第 0 个元素的位置 插入队列;
- ( 3 ) (3) (3) 从第 1 个元素开始迭代计算 d p [ i ] dp[i] dp[i];
- ( 4 ) (4) (4) 确保相邻两个元素的差值 ≤ k \le k ≤k;
- ( 5 ) (5) (5) O ( 1 ) O(1) O(1) 计算 d p [ i ] dp[i] dp[i];
- ( 6 ) (6) (6) 如果前面的都 小于 0,那么从第 i i i 个数开始肯定更优;
- ( 7 ) (7) (7) 确保队列单调递减;
- ( 8 ) (8) (8) 将当前元素插入队列;
- ( 9 ) (9) (9) 注意,这里需要遍历 d p [ i ] dp[i] dp[i] 统计最大值;
三、本题小知识
一般用单调队列来优化动态规划的问题中,都会涉及两次出队操作
( 1 ) (1) (1) 确保区间范围合法;
( 2 ) (2) (2) 确保队列的单调性;
四、加群须知
相信看我文章的大多数都是「 大学生 」,能上大学的都是「 精英 」,那么我们自然要「 精益求精 」,如果你还是「 大一 」,那么太好了,你拥有大把时间,当然你可以选择「 刷剧 」,然而,「 学好算法 」,三年后的你自然「 不能同日而语 」。
那么这里,我整理了「 几十个基础算法 」 的分类,点击开启:
如果链接被屏蔽,或者有权限问题,可以私聊作者解决。
大致题集一览:
为了让这件事情变得有趣,以及「 照顾初学者 」,目前题目只开放最简单的算法 「 枚举系列 」 (包括:线性枚举、双指针、前缀和、二分枚举、三分枚举),当有 一半成员刷完 「 枚举系列 」 的所有题以后,会开放下个章节,等这套题全部刷完,你还在群里,那么你就会成为「 夜深人静写算法 」专家团 的一员。
不要小看这个专家团,三年之后,你将会是别人 望尘莫及 的存在。如果要加入,可以联系我,考虑到大家都是学生, 没有「 主要经济来源 」,在你成为神的路上,「 不会索取任何 」。
🔥联系作者,或者扫作者主页二维码加群,加入刷题行列吧🔥
🔥让天下没有难学的算法🔥
C语言免费动漫教程,和我一起打卡! 🌞《光天化日学C语言》🌞
入门级C语言真题汇总 🧡《C语言入门100例》🧡
几张动图学会一种数据结构 🌳《画解数据结构》🌳
组团学习,抱团生长 🌌《算法入门指引》🌌
竞赛选手金典图文教程 💜《夜深人静写算法》💜