BZOJ 4551: [Tjoi2016&Heoi2016]树

4551: [Tjoi2016&Heoi2016]树

Time Limit: 20 Sec  Memory Limit: 128 MB
Submit: 748  Solved: 394
[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

HINT

新加数据9组(By HFLSyzx ),未重测--2016.8.2

Source

 

[Submit][Status][Discuss]

分析

先离线读入所有的操作，统计每个点的标记次数，用并查集构建每个点向上最近的标记点。倒序处理操作，如果是查询，直接输出并查集的答案即可；否则该点标记次数减1，如果刚好变成0，就把并查集连向其父亲结点。

代码

 #include <bits/stdc++.h>
 
 using namespace std;
 
 const int N = ;
 
 int n, m;
 
 int tot;
 int hd[N];
 int nt[N];
 int to[N];
 
 int ope[N];
 int num[N];
 int cnt[N];
 int ans[N];
 
 char oper[];
 
 int fa[N], f[N];
 
 int find(int u)
 {
     return fa[u] == u ? u : fa[u] = find(fa[u]);
 }
 
 void prework(int u, int fr)
 {
     f[u] = fr;
 
     if (cnt[u])
         fa[u] = u;
     else
         fa[u] = find(fr);
 
     for (int i = hd[u]; ~i; i = nt[i])
         if (to[i] != fr)prework(to[i], u);
 }
 
 signed main(void)
 {
     scanf("%d%d", &n, &m);
 
     memset(hd, -, sizeof(hd));
 
     for (int i = ; i < n; ++i)
     {
         int x, y; scanf("%d%d", &x, &y);
 
         nt[tot] = hd[x]; to[tot] = y; hd[x] = tot++;
         nt[tot] = hd[y]; to[tot] = x; hd[y] = tot++;
     }
 
     ++cnt[];
 
     for (int i = ; i <= m; ++i)
     {
         scanf("%s%d", oper, num + i);
 
         if (oper[] == 'C')
             ope[i] = , ++cnt[num[i]];
     }
 
     prework(, );
 
     for (int i = m; i >= ; --i)
     {
         if (ope[i])
         {
             --cnt[num[i]];
             if (!cnt[num[i]])
                 fa[num[i]] = find(f[num[i]]);
         }
         else
             ans[i] = find(num[i]);
     }
 
     for (int i = ; i <= m; ++i)
         if (!ope[i])printf("%d\n", ans[i]);
 }

BZOJ_4551.cpp

@Author: YouSiki