P1801 黑匣子_NOI导刊2010提高(06)

题目描述

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。

扫描二维码关注公众号,回复: 71339 查看本文章

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

……A(M)。

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

…u(N)。

输出格式:

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

输入输出样例

输入样例#1: 
7 4
3 1 -4 2 8 -1000 2
1 2 6 6
输出样例#1: 
3
3
1
2

说明

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

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

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

Solution:

  本题是一道对顶堆的题目,但还是安利一波权值线段树。。。

  权值线段树其实很简单,故名思义它的每个叶子节点都有权值(即每个叶子节点维护的是某个确定的数),它可以很轻松的求出整个区间中的第$k$小的数(注意是整个区间,不是一段区间,权值线段树其实就是弱化的主席树,或者说主席树就是多棵权值线段树)。

  拿本题做例题。

  首先,我们对原数组离散化(因为数太大,空间开不下),这样离散后最多有$200000$个数,就能建树用叶子节点维护每个数了。

  再讲讲插入操作,其实就是单点修改,对于离散后的数每次找到它所在的叶子节点,使其权值$+1$(说明这个数出现了一次),并维护一段区间的数出现的个数(即区间求和)。

  然后询问第$k$小的数时,从整个区间$[1,n]$开始递归,当左儿子区间$[l,mid]$中出现的数个数大于等于$k$时,查询左儿子,否则查询右儿子中的第$k-sum[l,mid]$小的数(即左儿子中已有$sum[l,mid]<k$个数,那么整个区间第$k$小的数在右儿子里,且是右儿子中的第$k-sum[l,mid]$小的数),当查询到$l==r$时返回$l$就$ok$了。

代码:

 1 #include<bits/stdc++.h>
 2 #define il inline
 3 #define ll long long
 4 #define lson l,m,rt<<1
 5 #define rson m+1,r,rt<<1|1
 6 using namespace std;
 7 const int N=2e5+7;
 8 int tr[N<<2],n,m,u[N],s[N],cnt=1;
 9 struct numm{
10     ll v;int id;
11     bool operator < (numm a){return v<a.v;}
12 }a[N];
13 il ll gi(){
14     ll a=0;char x=getchar();bool f=0;
15     while((x>'9'||x<'0')&&x!='-')x=getchar();
16     if(x=='-')x=getchar(),f=1;
17     while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
18     return f?-a:a;
19 }
20 il void pushup(int rt){tr[rt]=tr[rt<<1]+tr[rt<<1|1];}
21 il void update(int u,int l,int r,int rt){
22     if(l==u&&r==u){tr[rt]++;return;}
23     tr[rt]++;
24     int m=l+r>>1;
25     if(u<=m)update(u,lson);
26     else update(u,rson);
27     pushup(rt);
28 }
29 il int query(int k,int l,int r,int rt){
30     if(l==r)return l;
31     int m=l+r>>1;
32     if(tr[rt<<1]>=k)return query(k,lson);
33     else return query(k-tr[rt<<1],rson);
34 }
35 int main()
36 {
37     n=gi(),m=gi();
38     for(int i=1;i<=n;i++)a[i].v=gi(),a[i].id=i;
39     for(int i=1;i<=m;i++)u[i]=gi();
40     sort(a+1,a+n+1);
41     for(int i=1;i<=n;i++)s[a[i].id]=i;
42     for(int i=1;i<=n;i++){
43         update(s[i],1,n,1);
44         while(i==u[cnt]){
45             printf("%lld\n",a[query(cnt,1,n,1)].v);
46             ++cnt;
47         }
48     }
49     return 0;
50 }

猜你喜欢

转载自www.cnblogs.com/five20/p/8952347.html