hdu2430Beans(单调队列)

来源:互联网 发布:淘宝上的剃刀龟 编辑:程序博客网 时间:2024/05/20 20:17

看好几篇题解都不知道怎么做!先抄一遍代码再说,好像发现了点问题,构造了一个多小时的数据,再想,再想,懂啦!!!

———————————————————————-

题意:
a[1], a[2], …, a[n] 取连续的一段(至少取一个),使得
(a[i+1] + a[i+2] + … + a[j]) % p <= k, 并且 (a[i+1] + a[i+2] + … + a[j])最大。
先前缀和 sum[i] !
那么问题就变成了 (i < j) (sum[j]-sum[i]) % p <= k, 并且(sum[j] - sum[i])最大。
==》 (i < j)
if sum[j]%p>sum[i]%p:
sum[j]%p-sum[i]%p <= k, 并且(sum[j] - sum[i])最大。
else:
sum[j]%p-sum[i]%p+p <= k, 并且(sum[j] - sum[i])最大。
现在证明 else 的情况下 (i,j) 的组合不如 (0,j) 的组合。

i < j, sum[j]%p < sum[i]%p, sum[j]%p-sum[i]%p+p <=k
==》 sum[j]%p <= k+sum[i]%p-p
==》 sum[j]%p < k (因为 sum[i]%p-p<0)
所以 (0,j)组合满足 (sum[j]-sum[0])%p < k,并且有
sum[j]-sum[0] > sum[j]-sum[i].
证毕。
所以 答案的组合 (i,j) 一定满足 i < j,sum[i]%p < sum[j]%p,而且 j-i 越大越好。
按 sum[i]%p 排序,用单调队列维护排序前的序号递增,并且满足 sum[j]%p-sum[i]%p <= k。

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>using namespace std;struct Node{    int x,id;    bool operator < (const struct Node &b) const{    return (x==b.x&&id<b.id)||x<b.x;    }}A[1000005];int Q[1000005];long long sum[1000005];int main(void){    int t,_=0;    scanf("%d",&t);    while(++_<=t)    {        int n,p,k,rt=-1;        scanf("%d%d%d",&n,&p,&k);        for(int i=1;i<=n;i++)        {            int G;            scanf("%d",&G);            sum[i]=sum[i-1]+G;            A[i].id=i,A[i].x=sum[i]%p;        }        sort(A+1,A+n+1);        int hd=0,tl=0;        Q[tl++]=0;        for(int l=1;l<=n;l++)        {            while(hd<tl&&A[l].x-A[Q[hd]].x>k) ++hd;            while(hd<tl&&A[l].id<A[Q[tl-1]].id) --tl;            if(hd<tl)            rt=max(rt,(int)((sum[A[l].id]-sum[A[Q[hd]].id])/p));            Q[tl++]=l;        }        printf("Case %d: %d\n",_,rt);    }    return 0;}
0 0
原创粉丝点击