bzoj2733 离线+并查集+主席树

https://www.lydsy.com/JudgeOnline/problem.php?id=2733

网上清一色的合并线段树题解，我又不会，只能自己胡来，没想到Rush过去了

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

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

对于 %的数据 n≤,m≤n，q≤ 

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

Sample Input

Q
Q
B
B
Q
Q
Q
Sample Output
-

题意

这是一个只有建桥没有拆桥的操作，很显然可以离线重新排序，将之后会被建桥的区间连在一起，保证之后每次操作都是连续的一个区间。

然后发现就是维护一个区间K小，上主席树rush！rush！rush！就过了

#include <map>
#include <set>
#include <ctime>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <sstream>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
inline int read(){int now=;register char c=getchar();for(;!isdigit(c);c=getchar());
for(;isdigit(c);now=now*+c-'',c=getchar());return now;}
#define For(i, x, y) for(int i=x;i<=y;i++)
#define _For(i, x, y) for(int i=x;i>=y;i--)
#define Mem(f, x) memset(f,x,sizeof(f))
#define Sca(x) scanf("%d", &x)
#define Sca2(x,y) scanf("%d%d",&x,&y)
#define Sca3(x,y,z) scanf("%d%d%d",&x,&y,&z)
#define Scl(x) scanf("%lld",&x);
#define Pri(x) printf("%d\n", x)
#define Prl(x) printf("%lld\n",x);
#define CLR(u) for(int i=0;i<=N;i++)u[i].clear();
#define LL long long
#define ULL unsigned long long
#define mp make_pair
#define PII pair<int,int>
#define PIL pair<int,long long>
#define PLL pair<long long,long long>
#define pb push_back
#define fi first
#define se second
typedef vector<int> VI;
const double eps = 1e-;
const int maxn = 1e5 + ;
const int INF = 0x3f3f3f3f;
const int mod = 1e9 + ;
int N,M,K;
int val[maxn];
int nxt[maxn],ed[maxn];
int fa[maxn];
void init(){
    For(i,,N){
        ed[i] = i;
        nxt[i] = -;
        fa[i] = i;
    }
}
int find(int p){
    return p == fa[p]?p:fa[p] = find(fa[p]);
}
void Union(int p,int q){
    int fp = find(p),fq = find(q);
    if(fp == fq) return;
    nxt[ed[fp]] = fq;
    ed[fp] = ed[fq];
    fa[fq] = fp;
}
struct Node{
    int op,x,y;
    Node(int op = ,int x = ,int y = ):op(op),x(x),y(y) {}
}node[];
int Index[maxn],tot;
int T[maxn],lson[maxn * ],rson[maxn * ],c[maxn * ];
int build(int l,int r){
    int root = tot++;
    c[root] = ;
    if(l != r){
        int m = (l + r) >> ;
        lson[root] = build(l,m);
        rson[root] = build(m + ,r);
    }
    return root;
}
int update(int root,int pos,int val){
    int newroot = tot++,tmp = newroot;
    c[newroot] = c[root] + val;
    int l = ,r = N;
    while(l < r){
        int m = (l + r) >> ;
        if(pos <= m){
            lson[newroot] = tot++;
            rson[newroot] = rson[root];
            newroot = lson[newroot];
            root = lson[root];
            r = m;
        }else{
            rson[newroot] = tot++;
            lson[newroot] = lson[root];
            newroot = rson[newroot];
            root = rson[root];
            l = m + ;
        }
        c[newroot] = c[root] + val;
    }
    return tmp;
}
int query(int left_root,int right_root,int k){
    int l = ,r = N;
    while(l < r){
        int m = (l + r) >> ;
        if(c[lson[left_root]] - c[lson[right_root]] >= k){
            r = m;
            left_root = lson[left_root];
            right_root = lson[right_root];
        }else{
            l = m + ;
            k -= c[lson[left_root]] - c[lson[right_root]];
            left_root = rson[left_root];
            right_root = rson[right_root];
        }
    }
    return l;
}
int id[maxn];
PII st[maxn];
int POS[maxn];
int main()
{
    Sca2(N,M);
    For(i,,N){
        Sca(val[i]);
        id[val[i]] = i;
    }
    init();
    For(i,,M){
        Sca2(st[i].fi,st[i].se);
        Union(st[i].fi,st[i].se);
    }
    int K; Sca(K);
    For(i,,K){
        char op[]; scanf("%s",op);
        Sca2(node[i].x,node[i].y);
        if(op[] == 'B'){
            node[i].op = ;
            Union(node[i].x,node[i].y);
        }else{
            node[i].op = ;
        }
    }
    int cnt = ;
    For(i,,N){
        if(i == fa[i]){
            for(int j = i; ~j ; j = nxt[j]){
                Index[++cnt] = j;
                POS[j] = cnt;
            }
        }
    }
    T[N + ] = build(,N);
    for(int i = N; i ; i --){
        T[i] = update(T[i + ],val[Index[i]],);
    }
    init();
    For(i,,M) Union(st[i].fi,st[i].se);
    for(int i = ; i <= K ; i ++){
        if(node[i].op == ){
            Union(node[i].x,node[i].y);
        }else{
            int t = find(node[i].x);
            int l = POS[t];
            int r = POS[ed[t]];
            int k = node[i].y;
            //cout << l << "  " << r << "  " << k << endl;
            if(k > r - l +  || k < ){
                puts("-1");
                continue;
            }
            Pri(id[query(T[l],T[r + ],k)]);
        }
    }
    #ifdef VSCode
    system("pause");
    #endif
    return ;
}