HDU 4557 非诚勿扰 —— 优先队列+离散化+主席树

作为2013年699万应届毕业生中的一员，由于宏观经济的不景气，小明在毕业当天就华丽丽地失业了！
　　经历了千难万苦的求职过程，小明特别能理解毕业生的就业之难，所以，他现在准备创建一家专门针对IT人才的求职中介公司——非诚勿扰人力资源开发有限公司。
　　基于工作的需要，小明根据求职学生的简历描述为每人评定了一个综合能力值，能力值是一个小于等于20的正整数，值越高表示能力越强。当有公司试图招聘IT人员的时候（每次只招聘1名），需要提出一个综合能力的最低需求，若人才库中有符合要求的人才，则一定能成功招聘。当然，若有多名学生同时满足招聘公司的需求，鉴于高能力人才的稀缺，小明总是优先把能力值低的人才推荐过去；如果依然有多名人员符合要求，则小明就把其中最早来求职的那位学生推荐过去。
　　需要说明的是，刚开始的时候，公司的人才库为空，而且一名学生只能和一个企业签约，如果推荐成功，则该名学生的信息需要从人才库中删除。
Input
　　输入数据的第一行是一个正整数T（1 <= T <= 20）, 表示有T组测试数据；
　　每组测试数据第一行是一个整数n（0 <= n <= 1000），表示按照时间先后发生了n次事件。接下来的n行，每行描述一次事件。对于一次事件，先是一个字符串”Add”或者”Find”，其中”Add”表示有一名学生加入了人才库，”Find”表示有企业想招聘一名人员。
如果字符串是”Add”，则后面将有一个字符串s和一个数字d，用空格隔开，分别表示该名学生的名字和综合能力值，名字由小写字母组成，不为空且长度不超过15；如果字符串是”Find”，则后面将有一个数字，表示招聘公司对人才综合能力的最低要求。
Output
对于每组测试数据，第一行输出”Case #c:”（不包含引号）
c是测试数据的组数，从1开始。
然后输出n行，表示n次事件的结果
如果本次事件是添加人才信息入库，则请输出加入该信息后，人才库内的人员数量；
如果本次事件是企业来招聘，则请输出将被录用的人才名字，如果没有人才符合要求，就请输出”WAIT…”
Sample Input
1
5
Add lcy 1
Add lyd 19
Find 11
Find 13
Add zxs 10
Sample Output
Case #1:
1
2
lyd
WAIT…
2

这道题好像根本不需要主席树去做，因为是数据结构专场的题，就没想到别的，但是最终还是做出来了，虽然浪费一点时间。。。
用主席树做离线的题目也不是一次两次了（好像就两次），我本来想着在build的时候初始化的，然后发现是动态建点QAQ，只能for一遍初始化。所有输入之后先离散化，然后由于他的输入范围特别小，我们可以开1e3的优先队列，之后就是正常做了。

#include<bits/stdc++.h>
using namespace std;
#define inf 1e9
const int maxn=1e3+5;
int op[maxn],b[maxn],pos[maxn];
struct node
{
    int id;
    string s;
    bool operator< (const node& q)const
    {
        return id>q.id;
    }
    node(){}
    node(int id,string s):id(id),s(s){}
};
priority_queue<node>Q[maxn];
string s[maxn];
int minn[maxn*25],ls[maxn*25],rs[maxn*25],rt[maxn],cnt,have[maxn*25];
void pushup(int root)
{
    minn[root]=min(minn[ls[root]],minn[rs[root]]);
}
void update(int l,int r,int root,int last,int val,int pos)
{
    ls[root]=ls[last];
    rs[root]=rs[last];
    have[root]=have[last]+val*pos;
    if(l==r)
    {
        have[root]+=val*pos;
        if(have[root]==0)
            minn[root]=inf;
        else
            minn[root]=pos;
        return ;
    }
    int mid=l+r>>1;
    if(mid>=pos)
        update(l,mid,ls[root]=++cnt,ls[last],val,pos);
    else
        update(mid+1,r,rs[root]=++cnt,rs[last],val,pos);
    pushup(root);
}
int query(int l,int r,int root,int ql,int qr)
{
    if(minn[root]>=ql)
        return minn[root];
    if(minn[root]==inf)
        return inf;
    if(l>=ql&&r<=qr)
        return minn[root];
    int mid=l+r>>1;
    int ans=inf;
    if(mid>=ql)
        ans=min(ans,query(l,mid,ls[root],ql,qr));
    if(ans==inf&&mid<qr)
        ans=min(ans,query(mid+1,r,rs[root],ql,qr));
    return ans;
}
int main()
{
    int t;
    cin>>t;
    int cas=0;
    while(t--)
    {
        int n;
        cin>>n;
        //init
        for(int i=1;i<=n;i++)
            while(!Q[i].empty())
                Q[i].pop();
        cnt=0;
        for(int i=0;i<=n*25;i++)
            minn[i]=inf,have[i]=0;
        string ss;
        for(int i=1;i<=n;i++)
        {
            cin>>ss;
            if(ss[0]=='A')
                op[i]=1,cin>>s[i]>>pos[i];
            else
                op[i]=2,cin>>pos[i];
            b[i]=pos[i];
        }
        sort(b+1,b+1+n);
        int all=unique(b+1,b+1+n)-b-1;
        for(int i=1;i<=n;i++)
            pos[i]=lower_bound(b+1,b+all+1,pos[i])-b;
        int ans=0;
        int now,last=1;
        cout<<"Case #"<<++cas<<":"<<endl;
        for(int i=1;i<=n;i++)
        {
            if(op[i]==1)
            {
                Q[pos[i]].push(node(i,s[i])),ans++;
                now=++cnt;
                update(1,all,now,rt[last],1,pos[i]);
                last++;
                rt[last]=now;
                cout<<ans<<endl;
            }
            else
            {
                int fin=query(1,all,rt[last],pos[i],all);
                if(fin==inf||pos[i]>all)
                    cout<<"WAIT..."<<endl;
                else
                {
                    ans--;
                    node v=Q[fin].top();
                    Q[fin].pop();
                    int now=++cnt;
                    update(1,all,now,rt[last],-1,fin);
                    last++;
                    rt[last]=now;
                    cout<<v.s<<endl;
                }
            }
        }
    }
    return 0;
}