[HNOI2012] 永无乡 题解

题意：

　　n个点，有加边操作，询问与某一点处于相同的联通块的点中权值第k大的点

思路：

　　对所有点建立一棵权值线段树，加边就配合并查集进行线段树合并

反思：

　　动态开点，权值线段树要用sum[g[x=find(x)]]（还是不够熟练），g为根。

代码：

 #include<cstdio>
 const int M=,N=;
 int sz,a[M],p[M],g[M],id[M],lc[N],rc[N],sum[N];
 char c[];

 int read()
 {
     int x=; char ch=getchar();
     while (ch< || ch>) ch=getchar();
     while (ch> && ch<) x=(x<<)+(x<<)+ch-,ch=getchar();
     return x;
 }

 int find(int x) { for (;x^p[x];x=p[x]=p[p[x]]); return x; }

 void add(int l,int r,int &k,int x)
 {
     if (!k) k=++sz; ++sum[k];
     if (l==r) return;
     int mid=l+r>>;
     if (x>mid) add(mid+,r,rc[k],x);
     else add(l,mid,lc[k],x);
 }

 int merge(int x,int y)
 {
     if (!x || !y) return x|y;
     lc[x]=merge(lc[x],lc[y]),rc[x]=merge(rc[x],rc[y]);
     sum[x]=sum[lc[x]]+sum[rc[x]];
     return x;
 }

 int ask(int l,int r,int cur,int k)
 {
     if (l==r) return l;
     int mid=l+r>>;
     if (sum[lc[cur]]<k) return ask(mid+,r,rc[cur],k-sum[lc[cur]]);
     else return ask(l,mid,lc[cur],k);
 }

 int main()
 {
     int n=read(),m=read(),i,x,y;
     for (i=;i<=n;++i) id[a[i]=read()]=i,p[i]=i;
     for (i=;i<=m;++i) p[find(read())]=find(read());
     for (i=;i<=n;++i) add(,n,g[find(i)],a[i]);
     for (m=read();m--;)
         if (scanf("%s",c),x=read(),y=read(),c[]=='Q')
             if (sum[g[x=find(x)]]<y) puts("-1");
             else printf("%d\n",id[ask(,n,g[x],y)]);
         else
             if ((x=find(x))^(y=find(y)))
                 if (sum[y]<sum[x]) p[y]=x,g[x]=merge(g[x],g[y]);
                 else p[x]=y,g[y]=merge(g[y],g[x]);
     return ;
 }