2018年东北农业大学春季校赛 L-why的天鹅(离线线段树+离散化）

链接： https://www.nowcoder.com/acm/contest/93/L
来源：牛客网

题目描述

你们wyh学长小时候住在河边，因为周围的生态环境非常好，所以经常会有天鹅浮在湖面上，每只天鹅都长得不一样，它们偶尔排成一排，偶尔分散开，偶尔也会去其他河畔，wyh学长为了统计它们的个数，编了一个程序赋予它们一个“萌”值，但是这些天鹅很不听话，一会儿会从别的地方游过来一两只，一会儿又会在统计过程中游走一两只，现在请你帮他完成统计任务。

输入描述:

共有T(T<=10)组数据，每组数据第一行为两个数 N, M (N，M <= 500000)，代表有N只天鹅和M次操作，接下来一行是N个数字，下面M行首先会输入一个字符串S，接着会有三类操作，如果S是“insert”，接着输入一个正整数a，代表插入一只“萌”值为a的天鹅，如果S是“delete”，接着输入一个正整数a，代表删除一只“萌”值为a的天鹅，如果S是“query”，接着输入一个正整数k，代表查询“萌”值第k大的天鹅。
萌值为[1,1000000000]，并且保证一定存在第k大

输出描述:

对应每次询问，输出询问结果。

/*题解:
坐标是1~1000000000,但是点最多有1000000,因此可以对点进行离散化。
代码：*/

#include<bits/stdc++.h>
using namespace std;
int a[1000005],op[500005],id[500005],b[1000005],c[500005];
struct node
{
    int left,right,mid;
    int num;
}tree[4000005];
void build(int l,int r,int rt)
{
    tree[rt].left=l;
    tree[rt].right=r;
    tree[rt].mid=(l+r)>>1;
    tree[rt].num=0;
    if(l==r)return;
    build(l,tree[rt].mid,rt<<1);
    build(tree[rt].mid+1,r,rt<<1|1);
}
void add(int L,int rt,int k)
{
    if(tree[rt].left==tree[rt].right)
    {
        tree[rt].num+=k;
        return;
    }
    if(L<=tree[rt].mid)add(L,rt<<1,k);
    else add(L,rt<<1|1,k);
    tree[rt].num=tree[rt<<1].num+tree[rt<<1|1].num;
}
int query(int rt,int k)
{
    if(tree[rt].left==tree[rt].right)
        return tree[rt].left;
    if(tree[rt<<1|1].num>=k)return query(rt<<1|1,k);
    else return query(rt<<1,k-tree[rt<<1|1].num);
}
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        int n,m;scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
            c[i]=a[i];
        }
        int tol=n-1;
        for(int i=0;i<m;i++)//先把所有的点存储起来
        {
            char t[22];scanf("%s%d",&t,&id[i]);
            if(t[0]=='q')op[i]=0;
            else if(t[0]=='i')op[i]=1;
            else op[i]=2;
            a[tol++]=id[i];
        }
        sort(a,a+tol);
        int k=1;
        for(int i=1;i<tol;i++)//排序后去重
            if(a[i]!=a[i-1])a[k++]=a[i];//a的值可以直接用下标映射
        //for(int i=0;i<k;i++)b[i]=i;//把a中的点映射到b
        build(1,k,1);
        for(int i=0;i<n;i++)
        {
            int d=lower_bound(a,a+k,c[i])-a+1;
            add(d,1,1);
        }
        for(int i=0;i<m;i++)
        {
            if(op[i]==0)
            {
                printf("%d\n",a[query(1,id[i])-1]);
            }
            else if(op[i]==1)
            {
                int d=lower_bound(a,a+k,id[i])-a+1;
                add(d,1,1);
            }
            else
            {
                int d=lower_bound(a,a+k,id[i])-a+1;
                add(d,1,-1);
            }
        }
    }
    return 0;
}