1. 程式人生 > >hdu 2430 Beans 單調佇列

hdu 2430 Beans 單調佇列

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<string>
#include<algorithm>
#include<queue>
#include<cmath>
using namespace std;
//題意:從num[1]~num[n]中取一段連續的序列,使得(num[i]+...+num[j])%p<=k且該序列的和最大化
//令ma=a%mod,mb=b%mod,如果ma<mb,那麼(a-b)%mod的值會大於ma
//所以要想(a-b)%mod小於ma,那麼ma要大於mb
//這裡使用單調佇列就是要維護一個點字首和sum[i]%p單調遞增
//在使得sum[i]%p是單調的前提下維護位置是遞增的,這樣才會使得隊尾元素sum[tail]與隊首元素sum[head]的差值最大,即區間和最大
const int maxn=1000005;
int num[maxn];
long long sum[maxn];
struct node
{
	int pos,mod;
	node(int p,int m)
		:pos(p),mod(m){}
	node(){}
}a[maxn],que[maxn];

bool cmp(const node& a,const node& b) 
{
	if(a.mod==b.mod) return a.pos<b.pos;
	return a.mod<b.mod;
}

int main()
{
	int Case;
	scanf("%d",&Case);
	for(int kase=1;kase<=Case;kase++)
	{
		
		int n,p,k;
		scanf("%d%d%d",&n,&p,&k);
		sum[0]=0;
		for(int i=1;i<=n;i++)
		{
			scanf("%d",&num[i]);
			sum[i]=sum[i-1]+num[i];
			a[i].mod=sum[i]%p;
			a[i].pos=i;
		}
		sort(a+1,a+1+n,cmp);
		long long ans=-1;
		int head=0,tail=0;
		for(int i=1;i<=n;i++)
		{
			if(tail>head&&que[tail-1].pos>a[i].pos)
				tail--;//這樣為了保證佇列中的點的pos是單調的,只有這樣,隊尾減去隊首的差值才會最大
			que[tail++]=a[i];
			while(tail>head&&a[i].mod-que[head].mod>k) head++;
			if(a[i].mod<=k) ans=max(ans,sum[a[i].pos]);
			else if(head<tail-1)
			{
				ans=max(ans,sum[a[i].pos]-sum[que[head].pos]);
				//cout<<a[i].pos<<endl;
			}
		}
		if(ans!=-1) ans=ans/p;
		printf("Case %d: %I64d\n",kase,ans);

	}
	
}