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