JZOJ 1844——数数

来源:互联网 发布:博时基金公司知乎 编辑:程序博客网 时间:2024/05/16 08:32

这里写图片描述


仍然是预处理sum[i]表示前i个数的和,只枚举右端点r问题就转换成了在sum[r-k..r-1]中找一个最小值。那么我们维护一个元素单调递增的队列每次处理完一个r后把队尾所有不小于sum[r]的元素删掉后把sum[r]放进队尾。那么每次只用把队头中元素在原数组中的下标

代码如下:

#include<algorithm>#include<cstdio>#include<cmath>#include<iostream>#define MAXN 110000#define INF 100000009using namespace std;long long sum[MAXN];int q[MAXN];int n,k,head,tail;long long ans;int main(){    scanf("%d%d",&n,&k);    ans=-INF;q[0]=0;sum[0]=0;    for (int i=1;i<=n;i++)    {        int x;        scanf("%d",&x);        sum[i]=sum[i-1]+x;        while (q[head]<i-k) head++;        while (sum[i]<=sum[q[tail]]&&tail>=head)tail--;        q[++tail]=i;        ans=max(ans,sum[i]-sum[q[head]]);    }    printf("%lld",ans);    return 0;}
原创粉丝点击