KPI HDU - 5249【权值线段树】

Problem Description：

  你工作以后, KPI 就是你的全部了. 我开发了一个服务，取得了很大的知名度。数十亿的请求被推到一个大管道后同时服务从管头拉取请求。让我们来定义每个请求都有一个重要值。我的KPI是由当前管道内请求的重要值的中间值来计算。现在给你服务记录，有时我想知道当前管道内请求的重要值得中间值。

Input：

有大约100组数据。
每组数据第一行有一个n(1≤n≤10000)，代表服务记录数。
接下来有n行，每一行有3种形式
“in x”: 代表重要值为x(0≤x≤109)的请求被推进管道。
“out”: 代表服务拉取了管道头部的请求。
"query: 代表我想知道当前管道内请求重要值的中间值. 那就是说，如果当前管道内有m条请求, 我想知道，升序排序后第 $floor(m/2)+1th$ 条请求的重要值.
为了让题目简单，所有的x都不同，并且如果管道内没有值，就不会有"out"和"query"操作。

Output：

对于每组数据，先输出一行
Case # i:
然后每一次"query"，输出当前管道内重要值的中间值。

思路：

由于数据是动态更新的，所以采用离线做法，先把所有的数据保存。因为求的是第 $floor(m/2)+1$小，因此可以采用权值线段树的做法。
$ACcode$：

#include <bits/stdc++.h>
using namespace std;
const int N=1e4+5;
int tree[N<<2],a[N],b[N];
deque<int>dq,cmd;
char ss[10];
void build(int l,int r,int rt)
{
    if(l==r)
    {
        tree[rt]=0;
        return;
    }
    int mid=(l+r)>>1;
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void update(int l,int r,int p,int val,int rt)
{
    if(l==r)
    {
        tree[rt]+=val;
        return;
    }
    int mid=(l+r)>>1;
    if(p<=mid)
        update(l,mid,p,val,rt<<1);
    else
        update(mid+1,r,p,val,rt<<1|1);
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
int query(int l,int r,int p,int rt)
{
    if(l==r)
        return l;
    int mid=(l+r)>>1;
    if(tree[rt<<1]>=p)
        return query(l,mid,p,rt<<1);
    else
        return query(mid+1,r,p-tree[rt<<1],rt<<1|1);
}
int main()
{
    int n,cot=0;
    while(scanf("%d",&n)!=EOF)
    {
        int cnt=0,p=0;
        while(!dq.empty())
            dq.pop_front();
        while(!cmd.empty())
            cmd.pop_front();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",ss);
            if(strcmp(ss,"in")==0)
            {
                scanf("%d",&a[++cnt]);
                b[cnt]=a[cnt];
                cmd.push_back(1);
            }
            else if(strcmp(ss,"out")==0)
                cmd.push_back(2);
            else if(strcmp(ss,"query")==0)
                cmd.push_back(3);
        }
        sort(b+1,b+1+cnt);
        int len=unique(b+1,b+1+cnt)-b-1;
        build(1,len,1);
        printf("Case #%d:\n",++cot);
        while(!cmd.empty())
        {
            int op=cmd.front();
            cmd.pop_front();
            if(op==1)
            {
                int t=lower_bound(b+1,b+1+len,a[++p])-b;
                dq.push_back(a[p]);
                update(1,len,t,1,1);
            }
            else if(op==2)
            {
                int m=dq.front();
                dq.pop_front();
                int t=lower_bound(b+1,b+1+len,m)-b;
                update(1,len,t,-1,1);
            }
            else if(op==3)
            {
                int m=dq.size()/2+1;
                printf("%d\n",b[query(1,len,m,1)]);
            }
        }
    }
    return 0;
}