bzoj4551[Tjoi2016&Heoi2016]树

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

posted @ 2016-08-16 23:06  YuanZiming  阅读(181)  评论(0编辑  收藏  举报