洛谷3224 【HAOI2012】永无乡（线段树合并）

传送门

【题目分析】

线段树合并裸题了吧。。。。

将每个岛视作一棵权值线段树，然后直接按照题目意思去进行合并，由于只有查询第K大岛的询问，所以merge过程就直接递归到叶子节点，一直合并size即可。

第一次写线段树合并结果有一个地方写错调了半天，唉。。。

【代码~】

#include<bits/stdc++.h>
using namespace std;
const int MAXN=1e5+10;

int n,m,q;
int a[MAXN];
int tot,fa[MAXN],id[MAXN];
int lc[MAXN*20],rc[MAXN*20],siz[MAXN*20],rt[MAXN*20];

int Read(){
	int i=0,f=1;
	char c;
	for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
	if(c=='-')
	  f=-1,c=getchar();
	for(;c>='0'&&c<='9';c=getchar())
	  i=(i<<3)+(i<<1)+c-'0';
	return i*f;
}

int find(int x){
	if(x==fa[x])
	  return x;
	return fa[x]=find(fa[x]);
}

void update(int &root,int l,int r,int k){
	if(!root)
	  root=++tot;
	siz[root]++;
	if(l==r)
	  return ;
	int mid=l+r>>1;
	if(k<=mid)
	  update(lc[root],l,mid,k);
	else
	  update(rc[root],mid+1,r,k);
}

void merge(int &x,int y){
	if(!x||!y){
		x+=y;
	  	return ;
	}
	siz[x]+=siz[y];
	merge(lc[x],lc[y]);
	merge(rc[x],rc[y]);
}

void build(){
	for(int i=1;i<=n;++i){
		fa[i]=i;
		a[i]=Read();
		id[a[i]]=i;
		update(rt[i],1,n,a[i]);
	}
	for(int i=1;i<=m;++i){
		int x=Read(),y=Read();
		int fx=find(x),fy=find(y);
		if(fx!=fy){
			if(siz[rt[fx]]>siz[rt[fy]])
			  fa[fx]=fy,merge(rt[fy],rt[fx]);
			else
			  fa[fy]=fx,merge(rt[fx],rt[fy]);
		}
	}
}

int query(int root,int l,int r,int k){
	if(l==r)
	  return l;
	int x=siz[lc[root]];
	int mid=l+r>>1;
	if(k<=x)
	  return query(lc[root],l,mid,k);
	else
	  return query(rc[root],mid+1,r,k-x);
}

int main(){
	n=Read(),m=Read();
	build();
	q=Read();
	while(q--){
		char cz[10];
		scanf("%s",cz);
		int x=Read(),y=Read();
		if(cz[0]=='B'){
			int fx=find(x),fy=find(y);
			if(fx!=fy){
				if(siz[rt[fx]]>siz[rt[fy]])
			  	  fa[fx]=fy,merge(rt[fy],rt[fx]);
				else
			  	  fa[fy]=fx,merge(rt[fx],rt[fy]);
			}
		}
		else{
			x=find(x);
			if(siz[rt[x]]<y)
			  puts("-1");
			else
			  cout<<id[query(rt[x],1,n,y)]<<'\n';
		}
	}
	return 0;
}