BZOJ4946 NOI2017蔬菜（贪心+堆）
容易想到一个费用流做法：将每种蔬菜拆成p种，对应p个过期时间，每一种向可以卖的时间连边，第一次卖的奖励算在最晚过期的一种里。对于天数动态加点。不过这样边数太多了，因为第i天能卖的第i-1天一定能卖，可以改成每一种只向过期时间连边然后第i天向第i-1天连边。这样就有60分了。但费用流没有什么优化空间了。
如果蔬菜不会过期的话，贪心做法非常显然。那么，考虑让时光倒流。这样蔬菜只会每天增加，每次贪心的选出价值最大的行了。对于奖励，拆成两种蔬菜就好。
考虑对于单次询问具体应该怎么做。用一个大根堆维护应该选哪些蔬菜。堆里蔬菜分为两类，一类是每天都能拿的，一类只能拿一次的。每次从堆顶拿出m个蔬菜，如果是每天都能拿的再恢复回去，注意考虑各种情况。
然后考虑多次询问。我们需要从前p天的答案倒推出前p-1天的答案。注意到第p天能选择的第p-1天一定能选择。并且前p-1天的最优选择应该包含在前p天的最优选择内，否则第p天的选择本来就更少没有理由丢掉之前的较优选择。那么从答案里去掉最小的几个使得剩下的不超过(p-1)m个就可以了。
感觉写的姿势并不对，于是变的异常难写和丑陋不堪。
#include<iostream> #include<cstdio> #include<cmath> #include<cstdlib> #include<cstring> #include<algorithm> #include<vector> #include<queue> #include<stack> using namespace std; int read() { int x=0,f=1;char c=getchar(); while (c<'0'||c>'9') {if (c=='-') f=-1;c=getchar();} while (c>='0'&&c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar(); return x*f; } #define N 100010 #define inf 1000000000 #define ll long long int n,m,k,cnt=0; ll tot=0,ans[N]; struct data{int v,c,x; }a[N<<1]; struct data2{int i,x; }Q[N]; struct data3 { int tim,cnt,val; bool operator <(const data3&a) const { return val<a.val; } }choose[N*11]; vector<data3> app[N],rec[N]; priority_queue<data3> q; stack<data3> undo; bool cmp(const data2&a,const data2&b) { return a.x>b.x; } int main() { #ifndef ONLINE_JUDGE freopen("bzoj4946.in","r",stdin); freopen("bzoj4946.out","w",stdout); const char LL[]="%I64d\n"; #else const char LL[]="%lld\n"; #endif n=read(),m=read(),k=read(); for (int i=1;i<=n;i++) a[i].v=read(),a[i+n].v=a[i].v+read(),a[i].c=read()-1,a[i+n].c=1,a[i].x=read(),a[i+n].x=0; n<<=1; for (int i=1;i<=k;i++) Q[i].i=i,Q[i].x=read(); sort(Q+1,Q+k+1,cmp); for (int i=1;i<=n;i++) if (a[i].x==0) { if (i<=(n>>1)) app[Q[1].x].push_back((data3){0,a[i].c,a[i].v}); else app[a[i-(n>>1)].x?min(Q[1].x,a[i-(n>>1)].c/a[i-(n>>1)].x+1):Q[1].x].push_back((data3){0,a[i].c,a[i].v}); } else { if (a[i].c-1ll*a[i].x*Q[1].x>0) app[Q[1].x].push_back((data3){0,a[i].c-1ll*a[i].x*Q[1].x,a[i].v}); else if (a[i].c%a[i].x) app[a[i].c/a[i].x+1].push_back((data3){0,a[i].c%a[i].x,a[i].v}); rec[min(a[i].c/a[i].x,Q[1].x)].push_back((data3){min(a[i].c/a[i].x,Q[1].x)+1,a[i].x,a[i].v}); } for (int i=Q[1].x;i;i--) { for (int j=0;j<app[i].size();j++) q.push(app[i][j]); for (int j=0;j<rec[i].size();j++) q.push(rec[i][j]); int t=m; while (t&&!q.empty()) { if (q.top().tim==0) { if (t>=q.top().cnt) ans[Q[1].i]+=1ll*q.top().cnt*q.top().val,t-=q.top().cnt,choose[++cnt]=q.top(); else {ans[Q[1].i]+=1ll*t*q.top().val;undo.push((data3){0,q.top().cnt-t,q.top().val});choose[++cnt]=(data3){0,t,q.top().val};t=0;} } else { int tmp=q.top().cnt*(q.top().tim-i); if (t>=tmp) { ans[Q[1].i]+=1ll*tmp*q.top().val; choose[++cnt]=(data3){0,tmp,q.top().val}; undo.push((data3){i,q.top().cnt,q.top().val}); t-=tmp; } else { ans[Q[1].i]+=1ll*t*q.top().val; if (t>=q.top().cnt) choose[++cnt]=(data3){0,q.top().cnt*(t/q.top().cnt),q.top().val}; undo.push((data3){q.top().tim-(t-1)/q.top().cnt-1,q.top().cnt,q.top().val}); if (t%q.top().cnt) choose[++cnt]=(data3){0,t%q.top().cnt,q.top().val},undo.push((data3){0,q.top().cnt-t%q.top().cnt,q.top().val}); t=0; } } q.pop(); } while (!undo.empty()) q.push(undo.top()),undo.pop(); } sort(choose+1,choose+cnt+1); int tot=0;for (int i=1;i<=cnt;i++) tot+=choose[i].cnt; int t=2;long long sum=ans[Q[1].i];cnt=1; for (int i=Q[1].x-1;i;i--) { while (tot>i*m) { if (tot-i*m>=choose[cnt].cnt) sum-=1ll*choose[cnt].cnt*choose[cnt].val,tot-=choose[cnt++].cnt; else sum-=1ll*choose[cnt].val*(tot-i*m),choose[cnt].cnt-=tot-i*m,tot=i*m; } if (i==Q[t].x) ans[Q[t++].i]=sum; } for (int i=1;i<=k;i++) printf(LL,ans[i]); return 0; }
