[BZOJ4530][Bjoi2014]大融合 LCT + 启发式合并

[BZOJ4530][Bjoi2014]大融合

试题描述

小强要在N个孤立的星球上建立起一套通信系统。这套通信系统就是连接N个点的一个树。

这个树的边是一条一条添加上去的。在某个时刻，一条边的负载就是它所在的当前能够

联通的树上路过它的简单路径的数量。

例如，在上图中，现在一共有了5条边。其中，(3,8)这条边的负载是6，因

为有六条简单路径2-3-8,2-3-8-7,3-8,3-8-7,4-3-8,4-3-8-7路过了(3,8)。

现在，你的任务就是随着边的添加，动态的回答小强对于某些边的负载的

询问。

输入

第一行包含两个整数N,Q，表示星球的数量和操作的数量。星球从1开始编号。

接下来的Q行，每行是如下两种格式之一：

A x y 表示在x和y之间连一条边。保证之前x和y是不联通的。

Q x y 表示询问(x,y)这条边上的负载。保证x和y之间有一条边。

1≤N,Q≤100000

输出

对每个查询操作，输出被查询的边的负载。

输入示例

8 6
A 2 3
A 3 4
A 3 8
A 8 7
A 6 5
Q 3 8

输出示例

6

题解

LCT + 启发式合并

我不敢直接用LCT直接维护有根树的子树size值，所以每次合并用启发式合并暴力将较小的连通块dfs重构（其实就是换根），再插到另一个连通块中。

当树Tree1的树根要作为另一棵树Tree2中节点u的儿子时，需要将Tree2中节点u到根节点的路径上每个节点的权值加上Tree1的大小，这是一个链上的问题，可以用LCT解决。（代码后附有更强的样例）

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <stack>
#include <vector>
#include <queue>
#include <cstdlib>
using namespace std;const int BufferSize = 1 << 16;
char buffer[BufferSize], *Head, *tail;
inline char Getchar() {if(Head == tail) {int l = fread(buffer, 1, BufferSize, stdin);tail = (Head = buffer) + l;}return *Head++;
}
int read() {int x = 0, f = 1; char c = Getchar();while(!isdigit(c)){ if(c == '-') f = -1; c = Getchar(); }while(isdigit(c)){ x = x * 10 + c - '0'; c = Getchar(); }return x * f;
}#define maxn 100010
#define maxm 200010
#define LL long long
int n, q, m, head[maxn], nxt[maxm], to[maxm];
void AddEdge(int a, int b) {to[++m] = b; nxt[m] = head[a]; head[a] = m;swap(a, b);to[++m] = b; nxt[m] = head[a]; head[a] = m;return ;
}int pa[maxn], siz[maxn];
int findset(int x) { return x == pa[x] ? x : pa[x] = findset(pa[x]); }int fa[maxn], ch[maxn][2], val[maxn], addv[maxn];
bool isroot(int u) { return ch[fa[u]][0] != u && ch[fa[u]][1] != u; }
void pushdown(int u) {int l = ch[u][0], r = ch[u][1];if(addv[u]) {addv[l] += addv[u]; addv[r] += addv[u];val[l] += addv[u]; val[r] += addv[u];addv[u] = 0;}return ;
}
void maintain(int u) {return ;
}
void rotate(int u) {int y = fa[u], z = fa[y], l = 0, r = 1;if(ch[y][1] == u) swap(l, r);if(!isroot(y)) ch[z][ch[z][1]==y] = u;fa[u] = z; fa[y] = u; fa[ch[u][r]] = y;ch[y][l] = ch[u][r]; ch[u][r] = y;maintain(y); maintain(u);return ;
}
int S[maxn], top;
void splay(int u) {S[++top] = u;for(int t = u; !isroot(t); t = fa[t]) S[++top] = fa[t];while(top) pushdown(S[top--]);while(!isroot(u)) {int y = fa[u], z = fa[y];if(!isroot(y)) {if((ch[y][0] == u) ^ (ch[z][0] == u)) rotate(u);else rotate(y);}rotate(u);}return ;
}
void access(int u) {for(int t = 0; u; u = fa[u]) {splay(u); ch[u][1] = t; maintain(u); t = u;}return ;
}
void add(int u, int v) {access(u); splay(u); addv[u] += v; val[u] += v;return ;
}
int query(int u) {access(u); splay(u);return val[u];
}
void rebuild(int u) {ch[u][0] = ch[u][1] = 0;val[u] = 1;for(int e = head[u]; e; e = nxt[e]) if(to[e] != fa[u]) {fa[to[e]] = u;rebuild(to[e]);val[u] += val[to[e]];}return ;
}int main() {n = read(); q = read();for(int i = 1; i <= n; i++) pa[i] = i, val[i] = siz[i] = 1;while(q--) {char tc = Getchar();while(!isalpha(tc)) tc = Getchar();int u = read(), v = read();if(tc == 'A') {int a = findset(u), b = findset(v);if(siz[a] > siz[b]) swap(a, b), swap(u, v);pa[a] = b; siz[b] += siz[a];fa[u] = v; AddEdge(u, v);rebuild(u);
//			printf("siz[%d] val[%d]: %d %d\n", b, u, siz[b], val[u]);add(v, siz[a]);}if(tc == 'Q') {LL x = (LL)min(query(u), query(v));
//			printf("%d %d\n", query(u), query(v));printf("%lld\n", x * (siz[findset(u)] - x));}}return 0;
}
/*
in:
8 14
A 2 3
Q 2 3
A 3 4
Q 2 3
A 3 8
Q 3 8
A 8 7
Q 3 4
A 6 5
Q 5 6
Q 3 8
A 1 6
A 1 8
Q 1 8
out:
1
2
3
4
1
6
15
*/