4946: [Noi2017]蔬菜

4946: [Noi2017]蔬菜

http://www.lydsy.com/JudgeOnline/upload/Noi2017D2.pdf

分析：

　　贪心。

　　首先可以将一个蔬菜拆成两个，一个是有加成的，一个是没有加成的。

　　贪心：1、多卖出些贵的好，所以先考虑贵的蔬菜；2、对于一个蔬菜，卖的越晚越好（越晚，可以给前面留出位置。）

　　然后对蔬菜按价格排序，从后往前考虑卖的时间，尽量卖。如果一天的m个蔬菜全卖了，那么下次走到这个位置就没用了，所以直接并查集合并即可。所以复杂度是$O(mn \times 并查集的复杂度)$。现在可以贪心的处理出任意天的最大买的获益。

　　如果对于每次询问都这样做，显然会超时。考虑优化。如果知道某一天，是否可以快速的知道相邻的一天。　

　　正着考虑，每次加入一些位置，因为上面的贪心是从最后一天贪心的，现在最后一天变了，所以无法推出下一天，没有什么关系。

　　正难则反，反过来考虑，每次相当于减去一些蔬菜，所以可以减去最便宜的。

　　做法：用上面的贪心处理处n天的最大获益，并记录每天卖了多少，用栈记录所有卖的蔬菜（上面的一定是价值小的），然后计算这一天需要减去多少颗蔬菜，从栈顶开始减。复杂度$O(n+D)$。

代码：

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<cmath>
 #include<iostream>
 #include<cctype>
 #include<set>
 #include<vector>
 #include<queue>
 #include<map>
 #define fi(s) freopen(s,"r",stdin);
 #define fo(s) freopen(s,"w",stdout);
 using namespace std;
 typedef long long LL;

 inline int read() {
     int x=,f=;char ch=getchar();for(;!isdigit(ch);ch=getchar())if(ch=='-')f=-;
     for(;isdigit(ch);ch=getchar())x=x*+ch-'';return x*f;
 }

 const int N = ;

 struct Node{
     int w, siz, sub, d; // 价值，数量，每天减少的，消失的一天
     Node() {}
     Node(int _w,int _siz,int _sub,int _d) { w = _w, siz = _siz, sub = _sub, d = _d; }
     bool operator < (const Node &A) const {
         return w > A.w;
     }
 }A[N];
 int n, m, k, Top;
 int sk[N], skw[N], d[N], fa[N], q[N];
 LL ans[N], sumr[N], del[N], Sum;

 int find(int x) {
     return x == fa[x] ? x : fa[x] = find(fa[x]);
 }
 void Merge(int a,int b) {
     a = find(a), b = find(b);
     if (a != b) fa[a] = b;
 }
 void Calc(int i,int s,int w) { // 第i天，卖价值为w的菜，卖了s个
     Sum += 1ll * s * w;
     d[i] += s;
     sk[Top] += s, skw[Top] = w; // skw[Top]+=w!!!
     if (d[i] == m) Merge(i, i - );
 }
 int main() {
     n = read(), m = read(), k = read();
     int tot = , D = ;
     for (int i=; i<=n; ++i) {
         int w = read(), fir = read(), siz = read(), sub = read();
         A[++tot] = Node(w + fir, , , sub ? (siz - ) / sub +  : N - );
         if (--siz) A[++tot] = Node(w, siz, sub, sub ? (siz - ) /sub +  : N - );
     }
     for (int i=; i<=k; ++i)
         q[i] = read(), D = max(q[i], D);

     sort(A + , A + tot + );
     for (int i=; i<=D; ++i) fa[i] = i;
     for (int i=; i<=tot; ++i) {
         ++Top;
         int idx = find(min(A[i].d, D));
         int sum = (idx - ) * A[i].sub, r = A[i].siz - sum; // sum前面卖的，r现在卖的
         while (idx && r) {
             int mn = min(m - d[idx], r); // d[idx] not d[idx - 1]
             Calc(idx, mn, A[i].w);
             r -= mn;
             int p = idx;
             idx = find(idx - );
             p -= idx;
             if (sum) r += p * A[i].sub, sum -= p * A[i].sub; //!!!
         }
         if (!find(D)) break;
     }
     for (int i=; i<=D; ++i) sumr[i] = sumr[i - ] + m - d[i]; // 剩余的前缀和
     LL tmp = ;
     for (int i=D; i>=; --i) del[i] = max(tmp - sumr[i], 0ll), tmp += d[i]; // 到第i天要删除的个数
     for (int i=D; i>=; --i) {
         ans[i] = Sum;
         int now = del[i - ] - del[i]; // 从第i天到第i-1天，要减去多少蔬菜。
         while (now) {
             int mn = min(now, sk[Top]);
             Sum -= 1ll * skw[Top] * mn;
             now -= mn;
             sk[Top] -= mn;
             if (!sk[Top]) Top --;
         }
     }
     for (int i=; i<=k; ++i) printf("%lld\n",ans[q[i]]);
     return ;
 }