bzoj4551 [Tjoi2016&Heoi2016]树

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

正解：并查集。

考虑离线倒序处理，先把标记结点和它的子树合并，然后处理询问时查询就输出值就好，打标记的话就是要把当前标记去掉，那么只要当前这个点没有标记了就把它和它父亲合并。

//It is made by wfj_2048~
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <vector>
#include <cmath>
#include <queue>
#include <stack>
#include <map>
#include <set>
#define inf (1<<30)
#define il inline
#define RG register
#define ll long long
  
using namespace std;
  
struct edge{ int nt,to; }g[2000010];
struct node{ int d,x; }q[1000010];
  
int head[1000010],fa[1000010],f[1000010],cnt[1000010],ans[1000010],n,Q,num;
char s[5];
  
il int gi(){
    RG int x=0,q=1; RG char ch=getchar(); while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
    if (ch=='-') q=-1,ch=getchar(); while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar(); return q*x;
}
  
il void insert(RG int from,RG int to){ g[++num]=(edge){head[from],to},head[from]=num; return; }
  
il int find(RG int x){ return f[x]==x ? f[x] : f[x]=find(f[x]); }
  
il void unionn(RG int u,RG int v){ f[find(u)]=find(v); return; }
  
il void dfs(RG int x,RG int p){
    fa[x]=p; RG int v;
    for (RG int i=head[x];i;i=g[i].nt){
    v=g[i].to; if (v==p) continue;
    dfs(v,x); if (!cnt[v]) unionn(v,x);
    }
    return;
}
  
il void work(){
    n=gi(),Q=gi(); RG int u,v,x; cnt[1]=1; f[n]=n;
    for (RG int i=1;i<n;++i) f[i]=i,u=gi(),v=gi(),insert(u,v),insert(v,u);
    for (RG int i=1;i<=Q;++i){
    scanf("%s",s); x=gi();
    if (s[0]=='C') q[i].d=1,q[i].x=x,cnt[x]++;
    if (s[0]=='Q') q[i].d=2,q[i].x=x;
    }
    dfs(1,0); RG int tot=0;
    for (RG int i=Q;i;--i){
    if (q[i].d==1){ cnt[q[i].x]--; if (!cnt[q[i].x]) unionn(q[i].x,fa[q[i].x]); }
    if (q[i].d==2) ans[++tot]=find(q[i].x);
    }
    for (RG int i=tot;i;--i) printf("%d
",ans[i]); return;
}
  
int main(){
    work();
    return 0;
}