BZOJ2733/LG3324 「HNOI2014」永无乡 权值线段树合并

问题描述

BZOJ2733

LG3224

---

题解

对于每个结点建立一棵权值线段树。

查询操作就去查询第 \(k\) 大，合并操作就合并两颗权值线段树。

并查集维护连通性。

同时 STO hkk,zcr,zzk

---

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
	x=0;char ch=1;int fh;
	while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
	if(ch=='-') ch=getchar(),fh=-1;
	else fh=1;
	while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
	x*=fh;
}

template <typename Tp>
void fr(Tp &x){
	char ch=1;
	while(ch!='Q'&&ch!='B') ch=getchar();
	if(ch=='Q') x=1;
	else x=2;
}

const int maxn=100003;

int n,m,T;
int rk[maxn];
int pre[maxn];

int rt[maxn],L[maxn*22],R[maxn*22],cnt;
int val[maxn*22],fa[maxn];

void pushup(int x){
	val[x]=val[L[x]]+val[R[x]];
}

int insert(int x,int l,int r,int p,int v){
	if(!x) x=++cnt;
	if(l==r){
		val[x]+=v;
		return x;
	}
	int mid=(l+r)>>1;
	if(p<=mid) L[x]=insert(L[x],l,mid,p,v);
	else R[x]=insert(R[x],mid+1,r,p,v);
	pushup(x);
	return x;
}

int merge(int p,int q,int l,int r){
	if(!p||!q) return p+q;
	if(l==r){
		val[q]+=val[p];return q;
	}
	int mid=(l+r)>>1;
	L[q]=merge(L[p],L[q],l,mid);
	R[q]=merge(R[p],R[q],mid+1,r);
	pushup(q);
	return q;
}

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

int query(int x,int l,int r,int k){
	if(!x&&k) return -1;
	if(r-l+1<k) return -1;
	if(l==r) return l;
	int mid=(l+r)>>1;
	if(!L[x]&&!R[x]&&k) return -1;
	if(L[x]&&val[L[x]]>=k) return query(L[x],l,mid,k);
	else return query(R[x],mid+1,r,k-val[L[x]]);
}

void cz1(){
	int x,k;read(x);read(k);
	int xx=find(x);
	int ppp=query(rt[xx],1,100000,k);
	printf("%d\n",pre[ppp]);
}

void cz2(){
	int x,y;read(x);read(y);
	int xx=find(x),yy=find(y);
	if(xx==yy) return;
	fa[xx]=yy;
	merge(rt[xx],rt[yy],1,100000);
}

int main(){
	read(n);read(m);
	for(int i=1;i<=n;i++) read(rk[i]),pre[rk[i]]=i;
	for(int i=1;i<=n;i++) rt[i]=++cnt;
	for(int i=1;i<=n;i++){
		fa[i]=i;
		insert(rt[i],1,100000,rk[i],1);
	}

//	#define debug1

	#ifdef debug1
		for(int i=1;i<=n;i++)
			printf("%d %d\n",query(rt[i],1,100000,1),query(rt[i],1,100000,2));
	#endif

	for(int i=1,x,y;i<=m;i++){
		read(x);read(y);
		int xx=find(x),yy=find(y);
		if(xx==yy) continue;
		fa[xx]=yy;
		merge(rt[xx],rt[yy],1,100000);
	}
	int T,op;read(T);
	while(T--){
		fr(op);
		if(op==1) cz1();
		else cz2();
	}
	return 0;
}