传送门
吐槽:题目上给的 testcase 太迷惑人了。。。
题目描述:
小Z有一片森林,含有N个节点,每个节点上都有一个非负整数作为权值。初始的时候,森林中有M条边。
小Z希望执行T个操作,操作有两类:
Q x y k
查询点x到点y路径上所有的权值中,第k小的权值是多少。此操作保证点x和点y连通,同时这两个节点的路径上至少有k个点。L x y
在点x和点y之间连接一条边。保证完成此操作后,仍然是一片森林。为了体现程序的在线性,我们把输入数据进行了加密。设lastans为程序上一次输出的结果,初始的时候lastans为0。
- 对于一个输入的操作
Q x y k
,其真实操作为Q x^lastans y^lastans k^lastans
。- 对于一个输入的操作
L x y
,其真实操作为L x^lastans y^lastans
。其中^运算符表示异或。请写一个程序來帮助小Z完成这些操作。
对于所有的数据,n,m,T<=8*10^4.
求解树上一条链的第K小模板题博客:Count on a tree
解题思路 :题目上要求强制在线,所以不能离线建树。我们需要在线维护树的信息。如果要我们合并两个棵树,我们可以将结点数量小的那棵树接在另一颗树上(为了保证复杂度)。我们对小的那棵树进行DFS,在DFS过程中进行修改ST表和维护线段树。思路实现了,剩下就是代码了。(注意细节)
代码:
#include<bits/stdc++.h>
using namespace std;
const int maxn = 8e4 + 5;
inline void read(int &num) {
int x = 0, f = 1;
char ch = ' ';
while (ch != '-' && (ch < '0' || ch > '9'))ch = getchar();
if (ch == '-')f = -1, ch = getchar();
while (ch >= '0' && ch <= '9')x = x * 10 + ch - '0', ch = getchar();
num = x * f;
}
struct node {
int ls, rs, cnt;
#define left(a, b) p[a].ls, p[b].ls, l, mid
#define right(a, b) p[a].rs, p[b].rs, mid + 1, r
} p[maxn * 1000];
int root[maxn], times;
struct Edge {
int e;
int p;
} load[maxn * 2];
int head[maxn], sign;
void add_edge(int s, int e) {
load[++sign] = Edge{e, head[s]};
head[s] = sign;
}
int value[maxn], grand[maxn][20], depth[maxn], N;
int fa[maxn], sizes[maxn];
///并查集维护数量
int find(int x) { return fa[x] == x ? x : fa[x] = find(fa[x]); }
void insert(int &now, int old, int l, int r, int x) {
now = ++times;
p[now] = p[old], p[now].cnt++;
if (l == r) return;
int mid = (l + r) >> 1;
if (x <= mid) insert(left(now, old), x);
else insert(right(now, old), x);
}
int query(int s, int e, int fa, int ga, int l, int r, int k) {
if (l == r) return l;
int x = p[p[s].ls].cnt + p[p[e].ls].cnt - p[p[fa].ls].cnt - p[p[ga].ls].cnt;
int mid = (l + r) >> 1;
if (k <= x) return query(p[s].ls, p[e].ls, p[fa].ls, p[ga].ls, l, mid, k);
else return query(p[s].rs, p[e].rs, p[fa].rs, p[ga].rs, mid + 1, r, k - x);
}
///重新构建,并且维护LCA和状态
void dfs(int s, int pre) {
grand[s][0] = pre;
for (int i = 1; i <= N; i++) grand[s][i] = grand[grand[s][i - 1]][i - 1];
depth[s] = depth[pre] + 1;
insert(root[s], root[pre], 1, (int) 1e9, value[s]);
for (int i = head[s], e; ~i; i = load[i].p) {
e = load[i].e;
if (e ^ pre) dfs(e, s);
}
}
int get_lca(int a, int b) {
if (depth[a] > depth[b]) swap(a, b);
for (int i = N; i >= 0; i--) if (depth[b] >= depth[a] && depth[grand[b][i]] >= depth[a]) b = grand[b][i];
for (int i = N; i >= 0; i--) if (grand[a][i] != grand[b][i]) { a = grand[a][i], b = grand[b][i]; }
return a == b ? a : grand[b][0];
}
void merge(int s, int e) {
///将结点少的树合并在结点多的树上
int fs = find(s), fe = find(e);
if (sizes[fs] > sizes[fe]) {
sizes[fs] += sizes[fe];
fa[fe] = fs;
dfs(e, s);
} else {
sizes[fe] += sizes[fs];
fa[fs] = fe;
dfs(s, e);
}
}
void init(int n) {
N = log2(n);
for (int i = 1; i <= n; i++) {
fa[i] = i, sizes[i] = 1;
head[i] = -1;
}
depth[0] = -1;
}
int main() {
char mark[3];
int testcase, n, m, op, s, e, k, lastans = 0;
read(testcase), read(n), read(m), read(op);
init(n);
for (int i = 1; i <= n; i++) {
read(value[i]);
///每个点状态的初始化
insert(root[i], root[0], 1, 1e9, value[i]);
}
for (int i = 1; i <= m; i++) {
scanf("%d %d", &s, &e);
add_edge(s, e), add_edge(e, s);
merge(s, e); ///合并两棵树
}
while (op--) {
scanf("%s", mark);
if (mark[0] == 'Q') {
read(s), read(e), read(k);
s ^= lastans, e ^= lastans, k ^= lastans;
int lca = get_lca(s, e);
printf("%d\n", lastans=query(root[s], root[e], root[lca], root[grand[lca][0]], 1, 1e9, k));
} else {
read(s), read(e);
s ^= lastans, e ^= lastans;
add_edge(s, e), add_edge(e, s);
merge(s, e);
}
}
return 0;
}