bzoj2733

并查集+treap+启发式合并

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<ctime>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#include<set>
#define rep(i,l,r) for(int i=(l);i<(r);i++)
#define clr(a,x) memset(a,x,sizeof(a))
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define mkp(a,b) make_pair((a),(b))
int read(){
    int ans=0,f=1;
    char c=getchar();
    while(!isdigit(c)){
        if(c=='-') f=-1;
        c=getchar();
    }
    while(isdigit(c)){
        ans=ans*10+c-'0';
        c=getchar();
    }
    return ans*f;
}
const int maxn=500009;
struct node{
    int s,v,p,r;
    node*ch[2];
    void maintain(){
        s=ch[0]->s+ch[1]->s+1;
    }
};
node pool[maxn<<2],*root[maxn],*pt=pool,*null;
int n,m,q,p[maxn],f[maxn];
node*newnode(int x,int cnt){
    pt->s=1;pt->v=x;pt->p=cnt;pt->r=rand();
    pt->ch[0]=pt->ch[1]=null;
    return pt++;
}
int find(int x){
    return f[x]==x?f[x]:f[x]=find(f[x]);
}
void rot(node*&o,int d){
    node*t=o->ch[d^1];
    o->ch[d^1]=t->ch[d];t->ch[d]=o;
    o->maintain();t->maintain();
    o=t;
}
void insert(node*&o,node*x){
    if(o==null){
        o=newnode(x->v,x->p);
    }else if(x->v<o->v){
        insert(o->ch[0],x);
        if(o->ch[0]->r>o->r) rot(o,1);
    }else{
        insert(o->ch[1],x);
        if(o->ch[1]->r>o->r) rot(o,0);
    }
    o->maintain();
}
int query(node*o,int x){
    int t=o->ch[0]->s;
    if(x==t+1) return o->p;
    if(x<t+1) return query(o->ch[0],x);
    return query(o->ch[1],x-t-1);
}
int X;
void join(node*o){
    if(o==null) return;
    join(o->ch[0]);
    insert(root[X],o);
    join(o->ch[1]);
}
void merge(int x,int y){
    int u=find(x),v=find(y);
    if(u==v) return;
    if(root[u]->s<root[v]->s) swap(u,v);
    f[v]=u;
    X=u;join(root[v]);
}
void init(){
    null=newnode(0,0);
    null->s=0;null->r=0;
    rep(i,1,n+1) f[i]=i;
    rep(i,1,n+1) root[i]=null;
}
int main(){
    n=read();m=read();
    init();
    rep(i,1,n+1){
        p[i]=read();p[i]=n-p[i]+1;
        insert(root[i],newnode(p[i],i));
    }
    rep(i,1,m+1){
        int from=read(),to=read();
        merge(from,to);
    }
    q=read();
    while(q--){
        char opt=getchar();
        while(opt!='Q'&&opt!='B') opt=getchar();
        int x=read(),y=read();
        if(opt=='Q') printf("%d
",y>root[find(x)]->s?-1:query(root[find(x)],root[find(x)]->s-y+1));
        else merge(x,y);
    }
    return 0;
}

2733: [HNOI2012]永无乡

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 1690  Solved: 891
[Submit][Status][Discuss]

Description

永无乡包含 n 座岛，编号从 1 到 n，每座岛都有自己的独一无二的重要度，按照重要度可 以将这 n 座岛排名，名次用 1 到 n 来表示。某些岛之间由巨大的桥连接，通过桥可以从一个岛 到达另一个岛。如果从岛 a 出发经过若干座（含 0 座）桥可以到达岛 b，则称岛 a 和岛 b 是连 通的。现在有两种操作：B x y 表示在岛 x 与岛 y 之间修建一座新桥。Q x k 表示询问当前与岛 x连通的所有岛中第 k 重要的是哪座岛，即所有与岛 x 连通的岛中重要度排名第 k 小的岛是哪 座，请你输出那个岛的编号。 
 

Input

输入文件第一行是用空格隔开的两个正整数 n 和 m，分别 表示岛的个数以及一开始存在的桥数。接下来的一行是用空格隔开的 n 个数，依次描述从岛 1 到岛 n 的重要度排名。随后的 m 行每行是用空格隔开的两个正整数 ai 和 bi，表示一开始就存 在一座连接岛 ai 和岛 bi 的桥。后面剩下的部分描述操作，该部分的第一行是一个正整数 q， 表示一共有 q 个操作，接下来的 q 行依次描述每个操作，操作的格式如上所述，以大写字母 Q 或B 开始，后面跟两个不超过 n 的正整数，字母与数字以及两个数字之间用空格隔开。 对于 20%的数据 n≤1000,q≤1000 
 
对于 100%的数据 n≤100000,m≤n，q≤300000 
 

Output

对于每个 Q x k 操作都要依次输出一行，其中包含一个整数，表 示所询问岛屿的编号。如果该岛屿不存在，则输出-1。 
 

Sample Input

5 1
4 3 2 5 1
1 2
7
Q 3 2
Q 2 1
B 2 3
B 1 5
Q 2 1
Q 2 4
Q 2 3

Sample Output

-1
2
5
1
2

HINT

 

Source

 

[Submit][Status][Discuss]