最大子序和
輸入一個長度爲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;
}