看好几篇题解都不知道怎么做!先抄一遍代码再说,好像发现了点问题,构造了一个多小时的数据,再想,再想,懂啦!!!
———————————————————————-
题意:
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;
}