bzoj2733: [HNOI2012]永无乡(并查集+主席树)

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/Hanks_o/article/details/79963178

题目传送门

解法:
跟3545解法差不多。
https://blog.csdn.net/hanks_o/article/details/79903342
这道题还没了边权。
直接连通。

那么并查集。
据说要启发式。
小的向大的合并。
然后就没了?

代码实现:

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
struct node {int lc,rc,c;}t[2100000];int cnt,rt[110000];
void build(int &u,int l,int r,int p) {
    if(u==0)u=++cnt;t[u].c++;
    if(l==r)return ;int mid=(l+r)/2;
    if(p<=mid)build(t[u].lc,l,mid,p);
    else build(t[u].rc,mid+1,r,p);
}
void Merge(int &u1,int u2) {
    if(u1==0) {u1=u2;return ;}if(u2==0)return ;t[u1].c+=t[u2].c;
    Merge(t[u1].lc,t[u2].lc);Merge(t[u1].rc,t[u2].rc);
}
int fa[110000];int findfa(int x) {if(fa[x]!=x)fa[x]=findfa(fa[x]);return fa[x];}
int s[110000];
int find(int u,int l,int r,int k) {
    if(t[u].c==0)return -1;
    if(l==r)return s[l];int mid=(l+r)/2,c=t[t[u].lc].c;
    if(k<=c)return find(t[u].lc,l,mid,k);
    else return find(t[u].rc,mid+1,r,k-c);
}
int main() {
    int n,m;scanf("%d%d",&n,&m);cnt=0;
    for(int i=1;i<=n;i++){int x;scanf("%d",&x);build(rt[i],1,n,x);s[x]=i;}
    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 xx=findfa(x),yy=findfa(y);
        if(xx!=yy) {
            if(t[rt[xx]].c<=t[rt[yy]].c) {fa[xx]=yy;Merge(rt[yy],rt[xx]);}
            else {fa[yy]=xx;Merge(rt[xx],rt[yy]);}
        }
    }
    int Q;scanf("%d",&Q);
    while(Q--) {
        char s[5];int x,y;scanf("%s%d%d",s+1,&x,&y);
        if(s[1]=='Q') printf("%d\n",find(rt[findfa(x)],1,n,y));
        else {
            int xx=findfa(x),yy=findfa(y);
            if(xx!=yy) {
                if(t[rt[xx]].c<=t[rt[yy]].c) {fa[xx]=yy;Merge(rt[yy],rt[xx]);}
                else {fa[yy]=xx;Merge(rt[xx],rt[yy]);}
            }
        }
    }
    return 0;
}
阅读更多
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页