hdu5249(权值线段树)

链接：点击打开链接

题意：有3种形式

  "in x": 代表重要值为x(0≤x≤109)的请求被推进管道。
  "out": 代表服务拉取了管道头部的请求。
  "query: 代表我想知道当前管道内请求重要值的中间值. 那就是说，如果当前管道内有m条请求, 我想知道，升序排序后第floor(m/2)+1th 条请求的重要值.

代码：

#include <map>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
struct node{
    int u,sig;
}a[10005];
int sum[40005];
map<int,int> mp,mp2;
map<int,int>::iterator ite;
void change(int L,int p,int l,int r,int rt){
    if(l==r){
        sum[rt]+=p;
        return;
    }
    int m=(l+r)>>1;
    if(L<=m)
    change(L,p,l,m,rt<<1);
    else
    change(L,p,m+1,r,rt<<1|1);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
int query(int p,int l,int r,int rt){
    if(l==r)
    return mp2[l];
    int m=(l+r)>>1;
    if(p<=sum[rt<<1])                           //往左右儿子走的时候要注意一下判断条件
    return query(p,l,m,rt<<1);
    else
    return query(p-sum[rt<<1],m+1,r,rt<<1|1);
}
int main(){                                     //主要就是按权值进行建线段树，跟线段树大体上是一样的
    char s[10];
    int n,i,j,id,op,num,cas=1;
    while(scanf("%d",&n)!=EOF){
        mp.clear();
        mp2.clear();
        id=1;
        for(i=1;i<=n;i++){
            scanf("%s",s);
            if(s[0]=='q')
            a[i].sig=0;
            else if(s[0]=='o'){
                a[i].sig=1;
                while(a[id].sig!=2)
                id++;
                a[i].u=a[id++].u;
            }
            else{
                scanf("%d",&num);
                a[i].sig=2;
                a[i].u=num;
                mp[num]=1;
            }
        }
        id=op=0;                                //值域比较大先离散化
        for(ite=mp.begin();ite!=mp.end();ite++){
            ite->second=(++id);
            mp2[id]=ite->first;
        }
        memset(sum,0,sizeof(sum));
        printf("Case #%d:\n",cas++);
        for(i=1;i<=n;i++){
            if(a[i].sig==2){
                op++;
                change(mp[a[i].u],1,1,id,1);
            }
            else if(a[i].sig==1){
                op--;
                change(mp[a[i].u],-1,1,id,1);
            }
            else{
                num=op/2+1;
                printf("%d\n",query(num,1,id,1));
            }
        }
    }
    return 0;
}
/*
6
in 874
query
out
in 24622
in 12194
query
*/