最大子序列和 (单调队列优化DP)

Posted 行码棋

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了最大子序列和 (单调队列优化DP)相关的知识,希望对你有一定的参考价值。

题目:

输入一个长度为 n 的整数序列,从中找出一段长度不超过 m 的连续子序列,使得子序列中所有数的和最大。

注意: 子序列的长度至少是 1。

思路:

要求一段连续序列的和使用前缀和的思想,区间 [ i , j ] [i,j] [i,j]的连续序列和为 s u m ( i ) − s u m ( j − 1 ) sum(i)-sum(j-1) sum(i)sum(j1),注意是 j − 1 j-1 j1,后面有用

所以只需要找到 s u m ( i ) − s u m ( j − 1 ) sum(i)-sum(j-1) sum(i)sum(j1)的最大值,且区间长度小于等于 m m m就可以

我们可以固定 s u m ( i ) sum(i) sum(i),只要找到区间小于等于m的 s u m ( j − k ) sum(j-k) sum(jk)的最小值即可,

故使用单调队列优化,维护区间长度小于等于m的 s u m ( j ) sum(j) sum(j)的最小值

因为我们算的是所有区间和中最小的值,开始的时候j-1就成0的位置了,所以初始时要加上一个0的元素,那么就可以使 t t = 0 tt = 0 tt=0而不是 − 1 -1 1

代码:

#include<bits/stdc++.h>
using namespace std;
const int N = 3e5+5;
int n,m;
int s[N],q[N];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        cin>>s[i];
        s[i] += s[i-1];
    }
    int hh = 0,tt = 0;//要初始化,加上一个0元素
    int ans = -1e9;
    for(int i=1;i<=n;i++)
    {
        while(hh<=tt&& i-m > q[hh]) hh++;
        ans = max(ans,s[i]-s[q[hh]]);//更新结果
        while(hh<=tt && s[q[tt]] >= s[i]) tt--;
        q[++tt] = i;
        // for(int i=hh;i<=tt;i++) cout<<q[i]<<" ";
        // cout<<endl;
    }
    cout<<ans<<'\\n';
    return 0;
}

以上是关于最大子序列和 (单调队列优化DP)的主要内容,如果未能解决你的问题,请参考以下文章

TYVJ 1305 最大子序和 题解 单调队列优化DP

Acwing -- 单调队列优化的DP问题

单调队列与DP

单调队列

最大子序列和(单调队列算法)

《算法竞赛进阶指南》0x59单调队列优化DP AcWing 299裁剪序列