最大子序和

题目描述

输入一个长度为n的整数序列，从中找出一段不超过m的连续子序列，使得整个序列的和最大。
例如：1,-3,5,1,-2,3
当m=4时，S=5+1-2+3=7
当m=2或m=3时，S=5+1=6 

输入

第一行两个数n,m
第二行有n个数，要求在n个数找到最大子序和 

输出

一个数，数出他们的最大子序和 

样例输入 

6 4
1 -3 5 1 -2 3

样例输出 

7

提示

30%满足 n,m<=100
50%满足 n,m<=3000
100%满足n,m<=300000

题解：采用单调队列，首先求数组的前缀和s[i],将问题转化成求出两个位置i，看，使得s[i]-s[k]最大，i-k<=m。

#include<bits/stdc++.h>
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1e6+7;
typedef long long ll;
typedef pair<int,int>PII;

int a[N];
int s[N];//储存a[i]的前缀和
int n,m;
deque<int>q;//定义了一个双向队列，方便求s[i]-s[k],i-k<=m
int ans=-inf;
int main(){
    cin >> n >> m;
    for(int i=1;i<=n;i++){
        cin >> a[i];
    }
    for(int i=1;i<=n;i++){
        s[i]=s[i-1]+a[i];//求前缀和
    }
    for(int i=1;i<=n;i++){
        while(!q.empty()&&q.front()<i-m) q.pop_front(); //确保队列中的数都在区间[i-m,i]中
        if(q.empty()) {
            ans=max(ans,s[i]);  
        }else {
            ans=max(ans,s[i]-s[q.front()]); //进行比较求最大值
        }
        while(!q.empty()&&s[q.back()]>=s[i]) q.pop_back(); //确保队列中的队首是队列中最小的元素并且每个元素都可以入队
        q.push_back(i);
    }
    cout << ans << "\n";
    return 0;
}