luogu P3224 [HNOI2012]永无乡 |线段树合并

题目描述

永无乡包含 (n) 座岛,编号从 (1)(n) ,每座岛都有自己的独一无二的重要度,按照重要度可以将这 (n) 座岛排名,名次用 (1)(n) 来表示。某些岛之间由巨大的桥连接,通过桥可以从一个岛到达另一个岛。如果从岛 (a) 出发经过若干座(含 (0) 座)桥可以 到达岛 (b) ,则称岛 (a) 和岛 (b) 是连通的。

现在有两种操作:

B x y 表示在岛 (x) 与岛 (y) 之间修建一座新桥。

Q x k 表示询问当前与岛 (x) 连通的所有岛中第 (k) 重要的是哪座岛,即所有与岛 (x) 连通的岛中重要度排名第 (k) 小的岛是哪座,请你输出那个岛的编号。

输入格式

第一行是用空格隔开的两个正整数 (n)(m) ,分别表示岛的个数以及一开始存在的桥数。

接下来的一行是用空格隔开的 (n) 个数,依次描述从岛 (1) 到岛 (n) 的重要度排名。随后的 (m) 行每行是用空格隔开的两个正整数 (a_i)​ 和 (b_i) ,表示一开始就存在一座连接岛 (a_i) 和岛 (b_i)​ 的桥。

后面剩下的部分描述操作,该部分的第一行是一个正整数 (q) ,表示一共有 (q) 个操作,接下来的 (q) 行依次描述每个操作,操作的 格式如上所述,以大写字母 (Q)(B) 开始,后面跟两个不超过 (n) 的正整数,字母与数字以及两个数字之间用空格隔开。

输出格式

对于每个 Q x k 操作都要依次输出一行,其中包含一个整数,表示所询问岛屿的编号。如果该岛屿不存在,则输出 (-1)


这题让我体验到了,cin读入字符有多慢。。

省选开始就要卡常数,还是要多用scanf


#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=1e5+5,M=5e6+5;
inline int read(){
	int x=0; char c=getchar();
	while(c<'0'||c>'9')c=getchar();
	while(c>='0'&&c<='9'){x=(x<<1)+(x<<3)+(c^48); c=getchar();}
	return x;
}
#define mid ((l+r)>>1)
int ls[M],rs[M],root[N],val[M],id[M],num;
void update(int &p,int l,int r,int &pos,int &d){
	if(!p)p=++num;
	if(l==r){ val[p]++; id[p]=d; return; }
	if(pos<=mid)update(ls[p],l,mid,pos,d);
	else update(rs[p],mid+1,r,pos,d);
	val[p]=val[ls[p]]+val[rs[p]];
}
void merge(int l,int r,int &u,int &v){
	if(!u||!v){ u=u|v; return; }
	if(l==r){ val[u]+=val[v]; id[u]|=id[v]; return; }
	merge(l,mid,ls[u],ls[v]);
	merge(mid+1,r,rs[u],rs[v]);
	val[u]=val[ls[u]]+val[rs[u]];
}
int n,m,fa[N];
inline int get(int x){
	return (fa[x]==x)?x:fa[x]=get(fa[x]);
}
int askth(int &p,int l,int r,int k){
	if(val[p]<k||!p)return 0;
	if(l==r)return id[p];
	if(k<=val[ls[p]])return askth(ls[p],l,mid,k);
	else return askth(rs[p],mid+1,r,k-val[ls[p]]);
}
signed main(){
	n=read(),m=read();
	for(int i=1,x;i<=n;i++){
		x=read(); fa[i]=i;
		update(root[i],1,n,x,i);
	}
	for(int i=1,x,y;i<=m;i++){;
		x=get(read()),y=get(read());
		fa[y]=x; merge(1,n,root[x],root[y]);
	}
	int q=read(),x,y;
	char ch[5];
	while(q--){
		scanf("%s",&ch);
		x=read(),y=read();
		if(ch[0]=='B'){
			x=get(x),y=get(y);
			fa[y]=x; merge(1,n,root[x],root[y]);
		}else{
			int ans=askth(root[get(x)],1,n,y);
			if(!ans)printf("-1
");
			else printf("%d
",ans);
		}
	}	
}
原文地址:https://www.cnblogs.com/naruto-mzx/p/13036076.html