【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

题解：

同bzoj3319.

只是把边修改改成了点修改。

代码：

#include<iostream>
#include<cstdio>
#define N 100010
using namespace std;
int point[N],f[N],fa[N],next[N<<1],v[N],ans[N],x,y,n,m,cnt;
struct edge{int st,en;}e[N<<1];
struct use{int x,kind;}q[N];
char ch[5];
void add(int x,int y){
  next[++cnt]=point[x];point[x]=cnt;
  e[cnt].st=x;e[cnt].en=y;
}
int find(int x){return x==f[x]?x:f[x]=find(f[x]);}
void dfs(int x){
  if (v[x]) f[x]=x;else f[x]=fa[x];
  for (int i=point[x];i;i=next[i])
    if (e[i].en!=fa[x]){fa[e[i].en]=x;dfs(e[i].en);}
}
int main(){
  scanf("%d%d",&n,&m);
  for (int i=1;i<n;i++){
    scanf("%d%d",&x,&y);
    add(x,y);add(y,x);
  } 
  for (int i=1;i<=m;i++){
    scanf("%s%d",&ch,&q[i].x);
    if (ch[0]=='C'){v[q[i].x]++;q[i].kind=1;}
  }
  v[1]++;dfs(1);
  for (int i=m;i>=1;i--){
    if (q[i].kind){
        v[q[i].x]--;
        if (v[q[i].x]==0) f[q[i].x]=fa[q[i].x];
    }
    else ans[i]=find(q[i].x);
  }
  for (int i=1;i<=m;i++) if (!q[i].kind)printf("%d\n",ans[i]);
}