BZOJ5132: [CodePlus2017年12月]火锅盛宴（洛谷P4032）

堆 线段树

BZOJ题目传送门
洛谷题目传送门

这道题有蛮多种做法的，但大体是堆+线段树/树状数组

我们把生的和熟的分开维护。生的用一个堆维护时间，每次读入时间后把熟了的转移到线段树上去，线段树维护区间和。

对于操作2，我的做法是另开n个队列维护每种食物被放入的先后顺序，当然树状数组+二分更快。

然后就差不多了，具体实现见代码：

#include<queue>
#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define M 500005
#define F inline
using namespace std;
struct que{
    int id,t;
    bool operator < (const que &a) const{
        return a.t<t;
    }
}tmp;
struct tree{ int l,r,x; }t[N<<2];
int n,s[N];
queue <que> a[N];
priority_queue <que> q;
F char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
F int _read(){
    int x=0; char ch=readc();
    while (!isdigit(ch)) ch=readc();
    while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
    return x;
}
F void writec(int x){ if (x>9) writec(x/10); putchar(x%10+48); }
F int _write(int x){ writec(x),putchar('\n'); return 1; }
void build(int x,int l,int r){
    t[x]=(tree){l,r,0};
    if (l==r) return; int mid=l+r>>1;
    build(x<<1,l,mid),build(x<<1|1,mid+1,r);
}
void mdfy(int x,int p,int w){
    t[x].x+=w; int mid=t[x].l+t[x].r>>1;
    if (t[x].l==t[x].r) return;
    p<=mid?mdfy(x<<1,p,w):mdfy(x<<1|1,p,w);
}
int srch_sum(int x,int l,int r){
    if (t[x].l>r||t[x].r<l) return 0;
    if (t[x].l>=l&&t[x].r<=r) return t[x].x;
    return srch_sum(x<<1,l,r)+srch_sum(x<<1|1,l,r);
}
int srch_id(int x){
    if (t[x].l==t[x].r) return t[x].l;
    t[x<<1].x?srch_id(x<<1):srch_id(x<<1|1);
}
int main(){
    for (int T=_read();T;T--){
        n=_read(),build(1,1,n);
        while (!q.empty()) q.pop();
        for (int i=1;i<=n;i++)
            while (!a[i].empty()) a[i].pop();
        for (int i=1;i<=n;i++) s[i]=_read();
        for (int m=_read();m;m--){
            int ti=_read(),f=_read(),id,l,r;
            while ((!q.empty())&&(q.top().t<=ti))
                id=q.top().id,mdfy(1,id,1),q.pop(),a[id].pop();
            switch (f){
                case 0: id=_read(),tmp=(que){id,ti+s[id]},q.push(tmp),a[id].push(tmp); break;
                case 1:
                    if (!srch_sum(1,1,n)) puts("Yazid is angry.");
                    else _write(id=srch_id(1)),mdfy(1,id,-1); break;
                case 2:
                    id=_read();
                    if (srch_sum(1,id,id)) mdfy(1,id,-1),puts("Succeeded!");
                    else if (a[id].empty()) puts("YJQQQAQ is angry.");
                    else _write(a[id].front().t-ti); break;
                case 3: l=_read(),r=_read(),_write(srch_sum(1,l,r)); break;
            }
        }
    }
    return 0;
}