BZOJ 4551 树

题目传送门

4551: [Tjoi2016&Heoi2016]树

Time Limit: 20 Sec Memory Limit: 128 MB

Submit: 508 Solved: 307

[Submit][Status][Discuss]

Description

在2016年，佳媛姐姐刚刚学习了树，非常开心。现在他想解决这样一个问题：给定一颗有根树（根为1），有以下
两种操作：1. 标记操作：对某个结点打上标记（在最开始，只有结点1有标记，其他结点均无标记，而且对于某个
结点，可以打多次标记。）2. 询问操作：询问某个结点最近的一个打了标记的祖先（这个结点本身也算自己的祖
先）你能帮帮他吗?

Input

输入第一行两个正整数N和Q分别表示节点个数和操作次数接下来N-1行，每行两个正整数u,v(1≤u,v≤n)表示u到v
有一条有向边接下来Q行，形如“opernum”oper为“C”时表示这是一个标记操作,oper为“Q”时表示这是一个询
问操作对于每次询问操作，1 ≤ N, Q ≤ 100000。

Output

输出一个正整数，表示结果

Sample Input

5 5
1 2
1 3
2 4
2 5
Q 2
C 2
Q 2
Q 5
Q 3

Sample Output

1
2
2
1

题解

此题很明显可以用线段树过，然而我比较懒，并不想打那么长的代码所以就用了一个离线算法配合并查集来过，还比较快。

#include <cstdio>
const int Maxn = 100005;
int marked[Maxn], father[Maxn], fa[Maxn], x[Maxn], ans[Maxn];
bool op[Maxn];

int find (int a) {
    return (father[a] == a) ? a : (father[a] = find(father[a]));
}

int main() {
    int N, Q, i, j, t;
    scanf("%d%d", &N, &Q);
    for (i = 1; i < N; ++i) {
        scanf("%d%d", &j, &t);
        fa[t] = j; father[t] = j;
    }
    father[1] = 1; marked[1] = 1;
    char S[2];
    for (i = 1; i <= Q; ++i) {
        scanf("%s%d", S, x + i);
        if (S[0] == 'C') ++marked[x[i]], father[x[i]] = x[i], op[i] = true;
}
    for (i = Q; i; --i)
      if (op[i]) {
          if (!(--marked[x[i]])) father[x[i]] = fa[x[i]];
      } else ans[i] = find(x[i]);
    for (i = 1; i <= Q; ++i)
      if (ans[i]) printf("%d\n", ans[i]);
    return 0;
}