hdu 5249(权值线段树）

一道权值线段树的题：也不算是板子题，变相求第k大

题目链接：KPI

题解：

由于数据范围比较大，所以需要离散化，二分+vector是我目前已知的最快的一种方法，这题就是一个变相求第k大问题，然后用队列模拟一下指令输入即可

AC代码：

#include<cstdio>
#include<algorithm>
using namespace std;
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
//求第k大
const int maxn=1e4+5;
int a[maxn];
vector<int>vec;
int getid(int x)//离散化
{
    return lower_bound(vec.begin(),vec.end(),x)-vec.begin()+1;
}
struct node
{
    int l,r,sum;
} tree[maxn*4];//这里需要注意一下，需要开大一点，不然就wa了
void pushup(int k)
{
    tree[k].sum=tree[k<<1].sum+tree[k<<1|1].sum;
}
void build(int k,int l,int r)
{
    tree[k].l=l;
    tree[k].r=r;
    tree[k].sum=0;
    if(l==r)
    {
        return ;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
}
void update(int k,int pos,int w)
{
    if(tree[k].l==tree[k].r)
    {
        tree[k].sum+=w;
        return ;
    }
    int mid=(tree[k].l+tree[k].r)>>1;
    if(mid>=pos)
        update(k<<1,pos,w);
    else
        update(k<<1|1,pos,w);
    pushup(k);
}
int query_kth(int k,int ik)
{
    if(tree[k].l==tree[k].r)
    {
        return tree[k].l;
    }
    if(tree[k<<1].sum>=ik)
        return query_kth(k<<1,ik);
    else
        return query_kth(k<<1|1,ik-tree[k<<1].sum);
}
int main()
{

    int n;
    char op[20],id=0;
    while(~scanf("%d",&n))
    {
        memset(a,0,sizeof(a));
        for(int i=1;i<=n;i++)
            vec.clear();
        queue<int>que;
        for(int i=1; i<=n; i++)
        {
            scanf("%s",op);
            if(op[0]=='i')
            {
                scanf("%d",&a[i]);
                vec.push_back(a[i]);
            }
            else if(op[0]=='o')
            {
                a[i]-=1;
            }
            else if(op[0]=='q')
            {
                a[i]-=2;
            }
        }
        sort(vec.begin(),vec.end());
        vec.erase(unique(vec.begin(),vec.end()),vec.end());
        //printf("%d\n",vec.size());
        build(1,1,vec.size());
        printf("Case #%d:\n",++id);
        for(int i=1; i<=n; i++)
        {
            if(a[i]>=0)
            {
                que.push(a[i]);
                update(1,getid(a[i]),1);
            }
            if(a[i]==-1)
            {
                int key=que.front();
                update(1,getid(key),-1);
                que.pop();
            }
            else if(a[i]==-2)
            {
                printf("%d\n",vec[query_kth(1,que.size()/2+1)-1]);
            }
        }
        //printf("%d\n",que.size());
    }
}