题目地址:https://www.luogu.org/problem/P1801

题目


Black Box是一种原始的数据库。它可以储存一个整数数组,还有一个特别的变量i。最开始的时候Black Box是空的.而i等于0。这个Black Box要处理一串命令。

命令只有两种:

ADD(x):把x元素放进BlackBox;

GET:i加1,然后输出Blackhox中第i小的数。

记住:第i小的数,就是Black Box里的数的按从小到大的顺序排序后的第i个元素。例如:

我们来演示一下一个有11个命令的命令串。(如下图所示)

现在要求找出对于给定的命令串的最好的处理方法。ADD和GET命令分别最多200000个。现在用两个整数数组来表示命令串:

1.A(1),A(2),…A(M):一串将要被放进Black Box的元素。每个数都是绝对值不超过2000000000的整数,M$200000。例如上面的例子就是A=(3,1,一4,2,8,-1000,2)。

2.u(1),u(2),…u(N):表示第u(j)个元素被放进了Black Box里后就出现一个GET命令。例如上面的例子中u=(l,2,6,6)。输入数据不用判错。

输入格式

第一行,两个整数,M,N。

第二行,M个整数,表示A(l)……A(M)。

第三行,N个整数,表示u(l)…u(N)。

输出格式

输出Black Box根据命令串所得出的输出串,一个数字一行。

输入输出样例

说明/提示

对于30%的数据,M≤10000;

对于50%的数据,M≤100000:

对于100%的数据,M≤200000。

 

ac代码:


 

#include <bits/stdc++.h>
using namespace std;
const int maxn = 4*200005;
int a[maxn], b[maxn], cnt[maxn], u[maxn];
int q, m, n;;
void update(int id )
{
    cnt[id] = cnt[id << 1] + cnt[id << 1 | 1];
}
void build(int l, int r, int id, int no)
{
    if(l == r)
    {
        cnt[id]++;
        return ;
    }
    int mid = (l + r) >> 1;
    if(no <= mid) build(l, mid, id << 1, no);// 单点
    else build(mid + 1, r, id << 1 | 1, no);
    update(id);
}
int query(int l, int r, int id, int k)
{
    if(l == r)
        return l; //返回1-n的第k大的数是去重之后的第几大数
    int mid = (l + r) >> 1;
    if(k <= cnt[id << 1]) return  query(l, mid, id << 1, k);
    else return  query(mid + 1, r, id << 1 | 1, k - cnt[id << 1]);//!k - cnt[id << 1]
}
int main()
{
    //freopen("/Users/zhangkanqi/Desktop/11.txt","r",stdin);
    scanf("%d %d", &m, &n);
    for(int i = 1; i <= m; i++)
    {
        scanf("%d", &a[i]);
        b[i] = a[i];
    }
    for(int i = 1; i <= n; i++)
        scanf("%d", &u[i]);
    sort(a+1, a+1+m);
    int num = unique(a+1, a+1+m) - (a+1);
    for(int i = 1; i <= n; i++)
    {
        for(int j = u[i - 1] + 1; j <= u[i]; j++)
        {
            int no = lower_bound(a+1, a+1+num, b[j]) - a;
            build(1, num, 1, no);
        }
        int id = query(1, num, 1, ++q);
        printf("%d\n", a[id]);
    }
    return 0;
}