题目描述
永无乡包含 (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);
}
}
}