SSL 2521 数数

来源:互联网 发布:明星淘宝店铺大全 编辑:程序博客网 时间:2024/04/30 19:35

题目大意:

    给出一个N个数的整数数列:A1,A2,A3……。选择长度不超过k的连续区间,使总和最大。

输入样例:

    3 2

    1 2 3

输出样例:

    5

 1<=k<=n<=10^5,Ai<=10^9


解题思路

    这题是一题单调队列的模板题,先记录第i个数的前缀和,我们维护一个元素单调递增的队列,每次处理完一个r后把队尾所有不小于sum[r]的元素删掉后把sum[r]放进队尾。那么每次只用把队头中元素在原数组中的下标<r-k的元素删掉,然后选取队头即可得到以r为右端点的最大总和。


辣鸡程序(对C++不熟):

#include<cstdio>
#define MAXN 100005
using namespace std;
long long q[MAXN], sum[MAXN];
int head,tail,i,n,x,k; 
long long ans;
int main()
{
    scanf("%d%d",&n,&k);
    head=1;tail=1;
    for (int i=1;i<=n;i++)
    {
      scanf("%d",&x);
      sum[i]=sum[i-1]+x;
      while (sum[i]<=sum[q[tail]]&&tail>=head) tail--;
      q[++tail]=i;
      while (q[head]<i-k) head++;
      if ((sum[i]-sum[q[head]])>ans) ans=sum[i]-sum[q[head]];
      }
    printf("%lld",ans);
    return 0;
}


PS 这是一道单调队列模板题

1 0
原创粉丝点击