[NOI2017]蔬菜

题面

题意

有n种蔬菜，每种蔬菜有c[i]的单位，每单位可以获得收益a[i]，第一次卖出它可以额外获利s[i]，每天会有x[i]个单位过期，每天最多卖出m个单位的蔬菜，k次询问，每次询问给出一个数p，询问用p天最多可以获利多少。

做法

首先考虑一次询问，则可以将时间倒序来看，这样蔬菜过期就可以看作是这一天将一部分的蔬菜运来，第一次卖也同样额外获利s，然后用优先队列贪心维护最大的m个即可。
现在考虑多次询问，首先可以发现这样一个结论：第i天一定可以卖第i+1天的蔬菜。考虑将第i+1天的答案转移到第i天，只要把第i+1天卖出的所有蔬菜中较小的几个蔬菜扔掉，即为第i天的答案，开个小根堆维护一下即可。

代码

#include<bits/stdc++.h>
#define ll long long
#define N 100100
#define MN 100000
using namespace std;

ll n,m,Q,lim,sum,top,a[N],s[N],c[N],x[N],sd[N],ans[N],sta[N];
struct Va
{
	ll val,id;
	bool operator < (const Va &u) const{return val<u.val;}
};
struct Vb
{
	ll val,id;
	bool operator < (const Vb &u) const{return val>u.val;}
};
vector<ll>ad[N];
priority_queue<Va>A;
priority_queue<Vb>B;

int main()
{
	ll i,j,t;
	cin>>n>>m>>Q;
	for(i=1;i<=n;i++)
	{
		scanf("%lld%lld%lld%lld",&a[i],&s[i],&c[i],&x[i]);
		if(x[i]) ad[min((ll)MN,(c[i]+x[i]-1)/x[i])].push_back(i);
		else ad[MN].push_back(i);
	}
	for(i=MN;i>=1;i--)
	{
		for(j=0;j<ad[i].size();j++)
		{
			t=ad[i][j];
			A.push((Va){s[t]+a[t],t});
		}
		lim=m;
		top=0;
		for(;lim&&!A.empty();)
		{
			Va tmp;
			tmp=A.top();
			A.pop();
			if(!sd[tmp.id])
			{
				sd[tmp.id]++;
				lim--;
				ans[MN]+=tmp.val;
				if(c[tmp.id]>1) A.push((Va){a[tmp.id],tmp.id});
			}
			else
			{
				t=min(c[tmp.id]-(i-1)*x[tmp.id]-sd[tmp.id],lim);
				lim-=t;
				sd[tmp.id]+=t;
				ans[MN]+=t*tmp.val;
				if(sd[tmp.id]<c[tmp.id]) sta[++top]=tmp.id;
			}
		}
		for(j=1;j<=top;j++) A.push((Va){a[sta[j]],sta[j]});
	}
	for(i=1;i<=n;i++)
	{
		sum+=sd[i];
		if(sd[i]>1) B.push((Vb){a[i],i});
		else if(sd[i]==1) B.push((Vb){a[i]+s[i],i});
	}
	for(i=MN-1;i>=0;i--)
	{
		ans[i]=ans[i+1];
		if(m*i>=sum) continue;
		lim=sum-m*i;
		for(;lim;)
		{
			Vb tmp;
			tmp=B.top();
			B.pop();
			if(sd[tmp.id]>1)
			{
				t=min(sd[tmp.id]-1,lim);
				sd[tmp.id]-=t;
				lim-=t;
				ans[i]-=tmp.val*t;
				sum-=t;
				if(sd[tmp.id]>1) B.push(tmp);
				else B.push((Vb){a[tmp.id]+s[tmp.id],tmp.id});
			}
			else
			{
				ans[i]-=tmp.val;
				lim--;
				sd[tmp.id]--;
				sum--;
			}
		}
	}
	while(Q--)
	{
		scanf("%lld",&t);
		printf("%lld\n",ans[t]);
	}
}