最大子序和:單調隊列維護一個上升序列

最大子序和

輸入一個長度爲n的整數序列,從中找出一段長度不超過m的連續子序列,使得子序列中所有數的和最大。

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

輸入格式
第一行輸入兩個整數n,m。
第二行輸入n個數,代表長度爲n的整數序列。
同一行數之間用空格隔開。

輸出格式
輸出一個整數,代表該序列的最大子序和。

數據範圍
1≤n,m≤300000
輸入樣例:
6 4
1 -3 5 1 -2 3
輸出樣例:
7

題目不難,我們容易想到暴力枚舉,其時間複雜度是O(n * m)的這顯然是不可取的。
再後我們容易想到用前綴和來枚舉暴力做法。
於是我們有了下面的推斷。將距離小於等於m的一系列前綴和拿出。

我們容易想到,這裏面的最大值一定是:最小的減去最大的,當然還要同時滿足最小的在最大的前面。
我們再想一想。這假設上述的條件滿足。
這兩個數之間有一個數是有一個數是大於或者等於這個最大值的,那麼這裏就矛盾了。最大的差值一定不可能是這兩個,因爲前面已經找到更大的數對了,所以我們可以維護一個區間隊列,這個區間隊列是升序的。

#include<iostream>
#include<queue>
#include<cstdio>
using namespace std;
typedef long long ll;
const int N = 3e5 + 10;
ll s[N], n, m;
int q[N];
int main() {
    scanf("%lld %lld", &n, &m);
    ll temp;
    s[0] = 0;
    for(ll i = 1; i <= n; i++) {
        scanf("%lld", &temp);
        s[i] = s[i - 1] + temp;
    }
    int h = 0, t = 0;
    ll ans = -0x3f3f3f3f;
    for(int i = 1; i <= n; i++) {
        if(i - q[h] > m)    h++;
        ans = max(ans, s[i] - s[q[h]]);
        while(h <= t && s[q[t]] >= s[i]) t--;//這裏保證隊列區間裏的數是上升的。如果不是上升,從隊尾刪除元素。知道上升爲止。
        q[++t] = i;
        // printf("%lld\n", ans);
    }
    printf("%lld", ans);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章