码迷,mamicode.com
首页 > 其他好文 > 详细

bzoj4551[Tjoi2016&Heoi2016]树

时间:2016-08-17 00:04:21      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:

bzoj4551[Tjoi2016&Heoi2016]树

题意:

给个根节点为1的n点树,初始时节点1标记,Q个操作,每次可以标记一个点或求一个点最近一个标记了的祖先。

题解:

链剖可以写,当正解应该是并查集。离线读入所有操作,累加每个节点的标记次数,之后所有未被标记的节点向其父亲节点连边,然后倒着来,如果操作是询问则输出这个节点在并查集中的根节点,如果是标记则将该节点的标记数减1,一旦这个节点的标记数减到了0,就让它向父亲节点连边。

代码:

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <algorithm>
 4 #define inc(i,j,k) for(int i=j;i<=k;i++)
 5 #define dec(i,j,k) for(int i=j;i>=k;i--)
 6 #define maxn 100100
 7 using namespace std;
 8 
 9 inline int read(){
10     char ch=getchar(); int f=1,x=0;
11     while(ch<0||ch>9){if(ch==-)f=-1; ch=getchar();} while(ch>=0&&ch<=9)x=x*10+ch-0,ch=getchar();
12     return f*x;
13 }
14 struct e{int t,n;}; e es[maxn*2]; int ess,g[maxn];
15 void pe(int f,int t){
16     es[++ess]=(e){t,g[f]}; g[f]=ess; es[++ess]=(e){f,g[t]}; g[t]=ess;
17 }
18 struct ask{char opt[3];int x;}; ask asks[maxn];
19 int n,fa[maxn],p[maxn],q,tag[maxn];
20 void dfs(int x,int f){for(int i=g[x];i;i=es[i].n)if(es[i].t!=f)fa[es[i].t]=x,dfs(es[i].t,x);}
21 int find(int x){return x==p[x]?x:p[x]=find(p[x]);}
22 void uni(int x){int y=find(fa[x]),z=find(x); if(y!=z)p[z]=y;}
23 int main(){
24     n=read(); q=read();
25     inc(i,1,n-1){int a=read(),b=read(); pe(a,b);} dfs(1,0); tag[1]++;
26     inc(i,1,q){scanf("%s",asks[i].opt); asks[i].x=read(); if(asks[i].opt[0]==C)tag[asks[i].x]++;}
27     inc(i,1,n)p[i]=i; inc(i,1,n)if(!tag[i])uni(i);
28     dec(i,q,1){
29         if(asks[i].opt[0]==C){
30             tag[asks[i].x]--;
31             if(!tag[asks[i].x])uni(asks[i].x);
32         }
33         if(asks[i].opt[0]==Q)asks[i].x=find(asks[i].x);
34     }
35     inc(i,1,q)if(asks[i].opt[0]==Q)printf("%d\n",asks[i].x);
36     return 0;
37 }

 

20160611

bzoj4551[Tjoi2016&Heoi2016]树

标签:

原文地址:http://www.cnblogs.com/YuanZiming/p/5778235.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!