2018.08.11 洛谷P3224 [HNOI2012]永无乡(线段树合并)

传送门
给出n个带点权的点,支持连边和查询连通块第k大。

这个貌似就是一道线段树合并的裸板啊。。。
代码:

#include<bits/stdc++.h>
#define N 100005
using namespace std;
inline int read(){
    int ans=0;
    char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
    return ans;
}
int n,m,q,a[N],fa[N*60],siz[N*60],rt[N*60],son[N*60][2],tot=0,mp[N];
inline int find(int x){return x==fa[x]?x:fa[x]=find(fa[x]);}
inline void update(int&p,int l,int r,int k){
    if(!p)p=++tot;
    ++siz[p];
    if(l==r)return;
    int mid=l+r>>1;
    if(k<=mid)update(son[p][0],l,mid,k);
    else update(son[p][1],mid+1,r,k);
}
inline void merge(int&x,int y){
    if(!x||!y){x+=y;return;}
    siz[x]+=siz[y];
    merge(son[x][0],son[y][0]);
    merge(son[x][1],son[y][1]);
}
inline int query(int p,int l,int r,int k){
    if(l==r)return l;
    int mid=l+r>>1;
    int tmp=siz[son[p][0]];
    if(tmp>=k)return query(son[p][0],l,mid,k);
    return query(son[p][1],mid+1,r,k-tmp);
}
int main(){
    n=read(),m=read();
    for(int i=1;i<=n;++i)a[i]=read(),fa[i]=i,mp[a[i]]=i,update(rt[i],1,n,a[i]);
    for(int i=1;i<=m;++i){
        int x=read(),y=read();
        int fx=find(x),fy=find(y);
        if(fx!=fy)fa[fx]=fy,merge(rt[fy],rt[fx]);
    }
    q=read();
    while(q--){
        char op[2];
        int x,y;
        scanf("%s%d%d",op,&x,&y);
        if(op[0]=='B'){
            int fx=find(x),fy=find(y);
            if(fx!=fy){
                fa[fx]=fy;
                merge(rt[fy],rt[fx]);
            }
        }
        else{
            int fx=find(x);
            printf("%d\n",siz[rt[fx]]<y?-1:mp[query(rt[fx],1,n,y)]);
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/dreaming__ldx/article/details/81591246
今日推荐