知识点 - 数据结构合并

知识点 - 数据结构合并

解决问题类型：

大致是一些不需要合并后重新拆开的题目？

复杂度：

$O(总共开的结点数)$

可以证明，因为每次调用合并函数，必定会减少一个结点

实现

Merge 函数是数据结构合并的关键函数，其一个特征就是不可逆。

int Merge(int u,int v){
    if(u==0||v==0)return u|v;
    ls[u]=Merge(ls[u],ls[v]);
    rs[u]=Merge(rs[u],rs[v]);
    if(ls[u]==0&&rs[u]==0){
        ///叶子结点，单独更新。
    }
    else {
        ///pushup
    }
   return u;
}

现在遇到的有线段树合并和字典树合并

线段树合并，是将所有权值线段树的信息都用一个Merge函数累加一下。

字典树也类似。

例题

HDU - 6191 Query on A Tree

离线一下，从叶子结点向上合并，只有当前结点所有的子树都被访问完了后，才对其进行询问操作。

可以在线操作，利用可持续化的思想

HYSBZ - 2733 永无乡

线段树合并模板题。

代码

#include <bits/stdc++.h>
using namespace std;
#define rep(i,j,k) for(int i = (int)j;i <= (int)k;i ++)
#define debug(x) cerr<<#x<<":"<<x<<endl
#define pb push_back
void test(){cerr<<"\n";}
template<typename T,typename... Args>void test(T x,Args... args){cerr<<x<<" ";test(args...);}

inline int rd() { int c = 0, f = 1;char ch = getchar();
    while (ch < '0' || ch > '9') {if (ch == '-') f = -1;ch = getchar();}
    while (ch >= '0' && ch <= '9') {c = c * 10 + ch - '0';ch = getchar();}
    return c * f;
}

typedef long long ll;
typedef pair<int,int> pi;
const int MAXN = (int)1e5+7;

int id[MAXN];
int fa[MAXN];
inline int findfa(int x) {
    if (x == fa[x]) return x;
    return fa[x] = findfa(fa[x]);
}
void unite(int x,int y) {
    int u = findfa(x),v = findfa(y);
    if (u == v) return;
    fa[v] = u;
}

int root[MAXN],ls[MAXN*20],rs[MAXN*20],sum[MAXN*20],tot;

inline void Update(int now,int p,int l,int r) {
    sum[now] ++;
    if (l == r) return;
    int m = l+r>>1;
    if (p <= m) Update(ls[now]=++tot,p,l,m);
    else        Update(rs[now]=++tot,p,m+1,r);
}

inline int Query(int now,int kth,int l,int r) {
    if (sum[now] < kth) return -1;
    if (l == r) return l;
    int m = l+r>>1;
    if (sum[ls[now]] >= kth) return Query(ls[now],kth,l,m);
    else      return Query(rs[now],kth-sum[ls[now]],m+1,r);

}

inline int Merge(int u,int v) {
    if (u == 0 || v == 0) return u|v;
    ls[u] = Merge(ls[u],ls[v]);
    rs[u] = Merge(rs[u],rs[v]);
    sum[u] = sum[ls[u]]+sum[rs[u]];
    return u;
}

int main()
{
    int N,M,Q;
    N = rd(),M = rd();
    rep(i,1,N) fa[i] = i;
    rep(i,1,N) {
        int tmp = rd();
        id[tmp] = i;
        Update(root[i]=++tot,tmp,1,N);
    }

    rep(i,1,M) {
        int x,y;
        x = rd();y = rd();
        int fx = findfa(x),fy = findfa(y);
        unite(x,y);
        root[fx] = Merge(root[fx],root[fy]);
    }
    Q = rd();
    while (Q --) {
        char op[10];int x,y;
        scanf("%s",op);x = rd(),y =rd();
        if (op[0] == 'B') {
            int fx = findfa(x),fy = findfa(y);
            unite(x,y);
            root[fx] = Merge(root[fx],root[fy]);
        }else {
            int fx = findfa(x);
            int res = Query(root[fx],y,1,N);
            if (res == -1) printf("-1\n");
            else           printf("%d\n",id[res]);
        }
    }
}