永无乡

永无乡
怎么感觉一道省选题比一道模板题还水

n < = 1 e 5 , 所 以 考 虑 线 段 树 合 并 , 然 后 本 题 直 接 对 每 个 点 建 线 段 树 就 做 完 了 n<=1e5,所以考虑线段树合并,然后本题直接对每个点建线段树就做完了 n<=1e5,线线

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

const int N=1e5+2;
int n,m,tnt=0;
int p[N],rt[N],ss[N];
int fa[N];
int find(int x){return fa[x]==x?x:fa[x]=find(fa[x]);} 
struct node{
	int sum,l,id,r;
}tree[N*21];
void push_up(int u){
	int ls=tree[u].l,rs=tree[u].r;
    tree[u].sum=tree[ls].sum+tree[rs].sum;
}
void add(int u,int k,int l,int r,int id){
	if(l==r){
		tree[u].id=id;
		tree[u].sum=1;
		return;
	}
	int mid=(l+r)>>1;
	if(k<=mid){
		if(!tree[u].l){
			tree[u].l=++tnt;
		}
		add(tree[u].l,k,l,mid,id);
    }
    else {
    	if(!tree[u].r){
    		tree[u].r=++tnt;
    	}
    	add(tree[u].r,k,mid+1,r,id);
    }
    push_up(u);
}
void merge(int u,int v,int l,int r){
	if(l==r){
		tree[u].sum+=tree[v].sum;
		return;
	}
	int mid=(l+r)>>1;
	if(tree[v].l)
	{
		if(!tree[u].l){
			tree[u].l=tree[v].l;
		}
		else merge(tree[u].l,tree[v].l,l,mid);
	}
	if(tree[v].r){
		if(!tree[u].r){
			tree[u].r=tree[v].r;
		}
		else merge(tree[u].r,tree[v].r,mid+1,r);
	}
	push_up(u);
}
void Union(int u,int v){
    int fu=find(u),fv=find(v);	
    if(fu!=fv){
    	merge(rt[fv],rt[fu],1,n);
    	fa[fu]=fv;
    	rt[fu]=fv;
	}
}
int query(int s,int k,int l,int r){
	if(l==r){
		return tree[s].id;
	}
	int ls=tree[s].l,rs=tree[s].r,mid=(l+r)>>1;
	if(tree[ls].sum>=k){
		return query(ls,k,l,mid);
	}
	else {
		return query(rs,k-tree[ls].sum,mid+1,r);
	}
}
int main(){
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++){scanf("%d",&p[i]);}
	for(int i=1;i<=n;i++) fa[i]=i,rt[i]=i;
	tnt=n;
	for(int i=1;i<=n;i++){
		add(rt[i],p[i],1,n,i);
	}
	for(int i=1;i<=m;i++){
		int u,v;
		scanf("%d%d",&u,&v);
        Union(u,v);
	}
	int QQ;
	scanf("%d",&QQ);
	while(QQ--){
		char op[2];
		scanf("%s",op);
		int x,y;
		scanf("%d%d",&x,&y);
		if(op[0]=='Q'){
			int fx=rt[fa[x]];
			if(tree[fx].sum<y){
				puts("-1");
				continue;
			}
			int ans=query(fx,y,1,n);
			printf("%d\n",ans);
		}
		if(op[0]=='B'){
			Union(x,y);
		}
	}
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值