[HNOI2012 永无乡]线段树+并查集

曾经的曾经,一名年轻的OIer以为这是一道可合并平衡树,然后暗骂了一句出题人。后来,他知道线段树居然是可以合并的。再后来,他又知道了线段树居然是可以动态开点的。年轻的少年回头一看发现:哇,原来这题是傻逼题。
没错,这个东西可以轻松线段树并查集搞掉,合并的时候由于是动态开的点,所以时间保障在log以内,之后权值线段树查询的时候分治搞一下就行了。
傻逼少年的代码

#include <cstdio>
using namespace std;
const int Maxn=2e6+5;
int n,m,q,a,b,cnt=0;
int w[Maxn],f[Maxn];
int root[Maxn],lson[Maxn],rson[Maxn],sum[Maxn],id[Maxn];
int fd(int x){
    if(x==f[x]) return f[x];
    else return f[x]=fd(f[x]);
}
void build(int &k,int l,int r,int val){
    if(!k) k=++cnt;
    if(l==r){ sum[k]=1; return; }
    int mid=(l+r)>>1;
    if(val<=mid) build(lson[k],l,mid,val);
    else build(rson[k],mid+1,r,val);
    sum[k]=sum[lson[k]]+sum[rson[k]];
} 
int merge(int x,int y){
    if(!x) return y;
    if(!y) return x;
    lson[x]=merge(lson[x],lson[y]);
    rson[x]=merge(rson[x],rson[y]);
    sum[x]=sum[lson[x]]+sum[rson[x]];
    return x;
}
int query(int o,int l,int r,int x){
    if(l==r) return l;
    int mid=(l+r)>>1;
    if(sum[lson[o]]>=x) return query(lson[o],l,mid,x);
    else return query(rson[o],mid+1,r,x-sum[lson[o]]);
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i) scanf("%d",&w[i]);
    for(int i=1;i<=n;++i) f[i]=i;
    for(int i=1;i<=m;++i){
        scanf("%d%d",&a,&b);
        int fx=fd(a),fy=fd(b);
        if(fx!=fy) f[fx]=fy;
    }
    for(int i=1;i<=n;++i){
        build(root[fd(i)],1,n,w[i]);
        id[w[i]]=i;
    }
    scanf("%d",&q);
    char s[3];
    for(int i=1;i<=q;++i){
        scanf("%s%d%d",s,&a,&b);
        if(s[0]=='Q'){
            int p=root[fd(a)];
            if(sum[p]<b) puts("-1");
            else{
                int ans=query(p,1,n,b);
                printf("%d\n",id[ans]);
            }
        }
        else{
            int fx=fd(a),fy=fd(b);
            if(fx!=fy){
                f[fx]=fy;
                root[fy]=merge(root[fx],root[fy]);
            }
        }
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值