[BZOJ4551][TJOI2016&&HEOI2016]树(并查集)

4551: [Tjoi2016&Heoi2016]树

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

 

DFS序+线段树直接做就好了,联赛T1难度。

这里讲下并查集做法,将操作倒序处理。初始时每个点若(最终状态)被标记则fa指向自己否则连向父亲,然后每次删除一个标记则将fa指向父亲。这里并查集并不是用来维护连通性,而是作为一个复杂度非常优的链表处理。

 1 #include<cstdio>
 2 #include<cstring>
 3 #include<algorithm>
 4 #define rep(i,l,r) for (int i=l; i<=r; i++)
 5 using namespace std;
 6 
 7 const int N=100010;
 8 int n,m,a,b,cnt,to[N<<1],nxt[N<<1],h[N],fa[N],f[N],d[N],ans[N],q[N];
 9 char str[N][3];
10 
11 void add(int a,int b){ to[++cnt]=b; nxt[cnt]=h[a]; h[a]=cnt; }
12 void dfs(int x){ for(int i=h[x]; i; i=nxt[i]) if(to[i]!=fa[x]) fa[to[i]]=x,dfs(to[i]); }
13 int find(int x){ return (f[x]==x)?x:(f[x]=find(f[x])); }
14 
15 int main(){
16     freopen("bzoj4551.in","r",stdin);
17     freopen("bzoj4551.out","w",stdout);
18     scanf("%d%d",&n,&m);
19     rep(i,1,n-1) scanf("%d%d",&a,&b),add(a,b),add(b,a);
20     dfs(1); d[1]=1;
21     rep(i,1,m){
22         scanf("%s%d",str[i],&q[i]);
23         if(str[i][0]=='C')  d[q[i]]++;
24     }
25     rep(i,1,n) f[i]=(d[i])?i:fa[i];
26     for(int i=m; i>=1; i--){
27         if(str[i][0]=='C'){
28             d[q[i]]--;
29             if(!d[q[i]]) f[q[i]]=fa[q[i]];
30         }
31         else ans[i]=find(q[i]);
32     }
33     rep(i,1,m) if(str[i][0]=='Q') printf("%d\n",ans[i]);
34     return 0;
35 }

 

posted @ 2018-03-16 10:15  HocRiser  阅读(209)  评论(0编辑  收藏  举报