HDU 3415 Max Sum of Max-K-sub-sequence

Posted 2020-11-25 zzhzzh123

题目链接

题目大意:

我们有一个环状的数列,内有n个元素,现在其中找一长度不超过k的连续子序列,使值最大.

分析:

原数组断环为链:

(A_1)..(A_n) = > (A_1).. (A_n A_{n+1})..(A_{n+k})

考虑一段连续的子序列的值就是前缀和相减的形式

我们得到前缀和数组

(S_1)..(S_{n+k})

考虑以i为右端点的子序列,共有k种,值分别是(S_i-S_{i-1}),(S_i-S_{i-2})...(S_i-S_{i-k})

显然我们要在这之中找一个最大值,即在(S_{i-1}),(S_{i-2})...(S_{i-k})中找一个最小值

问题转化成对于每个(S_i),在某一定长区间内找一最小值

ST表,线段树均可在(O(nlogn))的时间内解决

但是还可以更快,用单调队列

[Code]

#include <bits/stdc++.h>
using namespace std;

int read(){
    int x=0,flag=1; char c;
    for(c=getchar();!isdigit(c);c=getchar()) if(c=='-') flag=-1;
    for(;isdigit(c);c=getchar()) x=((x+(x<<2))<<1)+(c^48);
    return x*flag;
}

const int N=2e5+50;

int n,k;
int a[N];
int s[N];

int main() {
    int T=read();
    while(T--){
        n=read(),k=read();
        for(int i=1;i<=n;i++) a[i]=read();
        for(int i=n+1;i<=n+k+1;i++) a[i]=a[i-n];
        for(int i=1;i<=n+k+1;i++) s[i]=s[i-1]+a[i];
        
        int ans=-2e9;
        int l=-1,r=-1;
        
        deque<int> q;
        q.push_back(0);
        for(int i=1;i<=n+k+1;i++){
            while(!q.empty()&&i-q.front()>k) q.pop_front();
            if(s[i]-s[q.front()]>ans){
                ans=s[i]-s[q.front()];
                l=q.front()+1; r=i;
            }
            ans=max(ans,s[i]-s[q.front()]);
            while(!q.empty()&&s[q.back()]>=s[i]) q.pop_back();
            q.push_back(i);
        }

        if(l>n) l-=n;
        if(r>n) r-=n;
        
        printf("%d %d %d
",ans,l,r);
    }
    return 0;
}