hdu 2430 Beans 单调队列

标签：acm

#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);

	}
	
}

hdu 2430 Beans 单调队列

标签：acm

原文地址：http://blog.csdn.net/u014088857/article/details/44819761