题目地址: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;
}