BZOJ2733:[HNOI2012]永无乡

浅谈线段树合并:https://www.cnblogs.com/AKMer/p/10251001.html

题目传送门:https://lydsy.com/JudgeOnline/problem.php?id=2733

对每个联通块维护一个值域线段树,然后该合并合并该查询查询就好了。

时间复杂度:\(O(nlogn)\)

空间复杂度:\(O(nlogn)\)

代码如下:

#include <cstdio>
using namespace std;

const int maxn=1e5+5;

char s[5];
int n,m,q;
int fa[maxn],rk[maxn],id[maxn],rt[maxn];

int read() {
	int x=0,f=1;char ch=getchar();
	for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
	for(;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
	return x*f;
}

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

struct segment_tree {
	int tot;
	int sum[maxn*20],ls[maxn*20],rs[maxn*20];

	void update(int p) {
		sum[p]=sum[ls[p]]+sum[rs[p]];
	}

	void change(int &p,int l,int r,int pos) {
		if(!p)p=++tot;
		if(l==r) {sum[p]++;return;}
		int mid=(l+r)>>1;
		if(pos<=mid)change(ls[p],l,mid,pos);
		else change(rs[p],mid+1,r,pos);
		update(p);
	}

	int query(int p,int l,int r,int rk) {
		if(l==r)return l;
		int mid=(l+r)>>1;
		if(rk<=sum[ls[p]])return query(ls[p],l,mid,rk);
		else return query(rs[p],mid+1,r,rk-sum[ls[p]]);
	}

	int merge(int a,int b) {
		if(!a||!b)return a+b;
		ls[a]=merge(ls[a],ls[b]);
		rs[a]=merge(rs[a],rs[b]);
		update(a);return a;
	}
}T;

int main() {
	n=read(),m=read();
	for(int i=1;i<=n;i++)
		rk[i]=read(),id[rk[i]]=i,fa[i]=i;
	for(int i=1;i<=m;i++) {
		int a=find(read()),b=find(read());
		if(a!=b)fa[a]=b;
	}q=read();
	for(int i=1;i<=n;i++) {
		int a=find(i);
		T.change(rt[a],1,n,rk[i]);
	}
	for(int i=1;i<=q;i++) {
		scanf("%s",s+1);
		if(s[1]=='Q') {
			int a=find(read()),k=read();
			if(T.sum[rt[a]]<k) puts("-1");
			else printf("%d\n",id[T.query(rt[a],1,n,k)]);
		}
		else {
			int a=find(read()),b=find(read());
			if(a!=b) {
				fa[a]=b;
				rt[b]=T.merge(rt[b],rt[a]);
			}
		}
	}
	return 0;
}
posted @ 2019-01-10 19:51  AKMer  阅读(231)  评论(0编辑  收藏  举报