hdu3415 单调队列模板题

比较裸的单调队列

先求前缀和,枚举所有结束位置1~n+k即可

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 200005
using namespace std;

int ans,a[maxn],sum[maxn];

int main(){
    int T,N,K;
    scanf("%d",&T);
    while(T--){
        memset(sum,0,sizeof sum);
        scanf("%d%d",&N,&K);
        for(int i=1;i<=N;i++){
            scanf("%d",&a[i]);
            sum[i]=sum[i-1]+a[i];
        }
        for(int i=N+1;i<=N+1+K;i++){
            a[i]=a[i-N];
            sum[i]+=sum[i-1]+a[i];
        }
        
        ans=-1000000000;
        int ansl,ansr,l=1,r=1,q[maxn];//单调队列求长度为k的区间中的最小值 
        q[1]=0;sum[0]=0;
        for(int i=1;i<=N+K;i++){
            while(l<=r && q[l]<i-K)l++;
            if(ans<sum[i]-sum[q[l]]){
                ans=sum[i]-sum[q[l]];
                ansl=q[l]+1,ansr=i>N?i-N:i;
            }
            while(l<=r && sum[q[r]]>=sum[i]) r--;
            q[++r]=i;  
        }
        printf("%d %d %d\n",ans,ansl,ansr);
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/zsben991126/p/10195384.html
今日推荐