poj 2985 线段树求第k大的数

树状数组实现好了之后,改用线段树也实现了一下,接下来再用各种树实现,继续学习!!

View Code
#include<cstdio>
#include<cstring>
const int maxn = 222222;
#define mid (l+r)>>1
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
int sum[maxn<<2];//记录有几个点的范围在节点的范围内
int a[maxn],p[maxn];
int n;
/*void pushup(int rt){
sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
*///在这里 pushup操作不需要,因为更新操作是从上往下进行的,到哪个节点就更新哪个节点
void build(int l,int r,int rt){
if(l==1) sum[rt]=n;
else sum[rt]=0;
if(l==r) return ;
int m=mid;
build(lson);
build(rson);
}
void update(int l,int r,int rt,int val,bool flag){
if(!flag) sum[rt]--;
else sum[rt]++;
int m=mid;
if(l==r) return ;
if(val<=m) update(lson,val,flag);
else update(rson,val,flag);
// pushup(rt);
}
int find(int x){
return x==p[x] ? x : p[x]=find(p[x]);
}
void query(int l,int r,int rt,int k){
if(l==r) {
printf("%d\n",l);
return ;
}
int m=mid;
if(k<=sum[rt<<1|1]) query(rson,k);//右子树有大于等于k个数,第k个数肯定在右边
else query(lson,k-sum[rt<<1|1]);
}
int main(){
int i,m,q,x,y,k;
scanf("%d%d",&n,&m);
for(i=1;i<=n;i++) {
a[i]=1; p[i]=i;
}
build(1,n,1);
for(i=1;i<=m;i++) {
scanf("%d",&q);
if(q==0){
scanf("%d%d",&x,&y);
x=find(x);
y=find(y);
if(x==y) continue;
update(1,n,1,a[x],false);
update(1,n,1,a[y],false);
update(1,n,1,a[x]+a[y],true);
p[y]=x;
a[x]+=a[y];
}
else {
scanf("%d",&k);
query(1,n,1,k);
}
}
return 0;
}

 

posted @ 2011-12-26 08:33  Because Of You  Views(1990)  Comments(0Edit  收藏  举报