题面
题意
有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]);
}
}