4551: [Tjoi2016&Heoi2016]树

应该是和sdoi发生了同样的一道很好的想法题由于出题人没有想到某种做法变成了水题。
这道题显然可以nlog^2n的树链剖分,但是有一种很好的并查集做法:
用类似疯狂的馒头这道题的方法,维护father[i]表示离i最近的打了标记的祖先。把所有询问离线下来倒着枚举,如果某个操作是把点从标记变成不标记,那么把它的father变成它的父亲。
    
    
    
    
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
 
#define ll long long
#define inf 1e9
#define eps 1e-8
#define md
#define N 100010
using namespace std;
struct yts { int x,t,ne;} e[2*N];
struct QQ { int ty,x;} q[N];
int father[N],fa[N],v[N],c[N],ans[N];
int num;
char st[5];
void put(int x,int y)
{
num++; e[num].x=x; e[num].t=y;
e[num].ne=v[x]; v[x]=num;
}
int find(int x) { return father[x]==x?x:father[x]=find(father[x]);}
void dfs(int x)
{
if (c[x]) father[x]=x; else father[x]=fa[x];
for (int i=v[x];i;i=e[i].ne)
{
int y=e[i].t;
if (y!=fa[x]) { fa[y]=x; dfs(y);}
}
}
 
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
put(x,y); put(y,x);
}
c[1]=1;
for (int i=1;i<=m;i++)
{
scanf("%s%d",st,&q[i].x);
if (st[0]=='C') c[q[i].x]++,q[i].ty=1;
}
dfs(1);
for (int i=m;i;i--)
{
int x=q[i].x;
if (q[i].ty)
{
c[x]--; if (c[x]==0) father[x]=fa[x];
}
else ans[i]=find(x);
}
for (int i=1;i<=m;i++)
if (ans[i]) printf("%d\n",ans[i]);
return 0;
}

你可能感兴趣的:(4551: [Tjoi2016&Heoi2016]树)