BZOJ2733: [HNOI2012]永无乡

【传送门:BZOJ2733


简要题意:

  有n个点,每个点都有点权,m条边,给出q个操作,两种操作:

  B x y将x连向y

  Q x k求出所有与x相连通(包括x)的点中第k小的点的编号


题解:

  主席树+并查集

  将所有连通的点都并在一起,然后每次并查集合并的时候,合并能够代表这两个连通块的两棵主席树

  然后找第k小就行了,挺裸的


参考代码:

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
struct node
{
    int lc,rc,c;
}tr[2100000];int trlen,rt[110000];
void Link(int &u,int l,int r,int p)
{
    if(u==0) u=++trlen;
    tr[u].c++;
    if(l==r) return ;
    int mid=(l+r)/2;
    if(p<=mid) Link(tr[u].lc,l,mid,p);
    else Link(tr[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2)
{
    if(u1==0){u1=u2;return ;}
    if(u2==0) return ;
    tr[u1].c+=tr[u2].c;
    Merge(tr[u1].lc,tr[u2].lc);
    Merge(tr[u1].rc,tr[u2].rc);
}
int t[110000];
int findkth(int u,int l,int r,int k)
{
    if(l==r) return t[l];
    int c=tr[tr[u].lc].c,mid=(l+r)/2;
    if(k>tr[u].c) return -1;
    if(c>=k) return findkth(tr[u].lc,l,mid,k);
    else return findkth(tr[u].rc,mid+1,r,k-c);
}
int fa[110000];
int findfa(int x)
{
    if(fa[x]!=x) fa[x]=findfa(fa[x]);
    return fa[x];
}
int a[110000],s[110000],n;
int LS(int d)
{
    int l=1,r=n,ans;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(s[mid]<=d)
        {
            ans=mid;
            l=mid+1;
        }
        else r=mid-1;
    }
    return ans;
}
int main()
{
    int m;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),s[i]=a[i];
    sort(s+1,s+n+1);
    trlen=0;memset(rt,0,sizeof(rt));
    for(int i=1;i<=n;i++)
    {
        int c=LS(a[i]);
        t[c]=i;
        Link(rt[i],1,n,c);
    }
    for(int i=1;i<=n;i++) fa[i]=i;
    for(int i=1;i<=m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        int fx=findfa(x),fy=findfa(y);
        if(fx!=fy)
        {
            fa[fy]=fx;
            Merge(rt[fx],rt[fy]);
        }
    }
    int w;
    scanf("%d",&w);
    char st[3];
    for(int i=1;i<=w;i++)
    {
        int x,y;
        scanf("%s%d%d",st+1,&x,&y);
        if(st[1]=='Q') printf("%d\n",findkth(rt[findfa(x)],1,n,y));
        else
        {
            int fx=findfa(x),fy=findfa(y);
            if(fx!=fy)
            {
                fa[fy]=fx;
                Merge(rt[fx],rt[fy]);
            }
        }
    }
    return 0;
}

 

猜你喜欢

转载自www.cnblogs.com/Never-mind/p/8858393.html
今日推荐