题目大意:一棵树分为黑点和白点,可以改变一个点的颜色,问从1到v的路径上第一个黑点是哪个点
LCT做法:把黑点col记作1,sum[x]记录x点子树右多少个黑点,我们可以用LCT把1到x路径上所有的点组成一个splay,然后在splay上找关键字最小的点就可以了,不难发现这个题就算找路径上第k个点也是很容易实现的。
对于树链剖分的做法也很容易理解,就是剖分不停在向上爬的时候类似于二分查找去寻找第一个出现的黑点在哪里,col和sum数组就变成了线段树的端点修改。
两种方法的复杂度都是nlognlogn
#include <bits/stdc++.h>
using namespace std;
#define maxn 100010
inline int getint()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n,m,Q;
int next[maxn],tr[maxn][2],fa[maxn],col[maxn],sum[maxn];
bool rev[maxn];
vector<int>g[maxn];
bool isroot(int rt)
{
return tr[fa[rt]][0]!=rt && tr[fa[rt]][1]!=rt;
}
void pushup(int x)
{
sum[x]=sum[tr[x][0]]+sum[tr[x][1]]+col[x];
}
void pushdown(int x)
{
if(!isroot(x))
pushdown(fa[x]);
if(rev[x])
{
rev[x]^=1;rev[tr[x][1]]^=1;rev[tr[x][0]]^=1;
swap(tr[x][0],tr[x][1]);
}
}
void rotate(int &x)
{
int l,r,y,z;
y=fa[x];z=fa[y];
if(tr[y][0]==x) l=0;
else l=1; r=l^1;
if(!isroot(y))
{
if(tr[z][0]==y) tr[z][0]=x;
else tr[z][1]=x;
}
fa[x]=z;fa[y]=x;fa[tr[x][r]]=y;
tr[y][l]=tr[x][r];tr[x][r]=y;
pushup(y);pushup(x);
}
void splay(int x)
{
pushdown(x);
int y,z;
while(!isroot(x))
{
y=fa[x];z=fa[y];
if(!isroot(y))
{
if((tr[y][0]==x)^(tr[z][0]==y)) rotate(x);
else rotate(y);
}
rotate(x);
}
}
void access(int x)
{
int t=0;
while(x)
{
splay(x);
tr[x][1]=t;
pushup(x);
t=x;x=fa[x];
}
}
void rever(int x)
{
access(x);splay(x);rev[x]^=1;
}
void link(int x,int y)
{
rever(x);fa[x]=y;splay(x);
}
void cut(int x,int y)
{
rever(x);access(y);splay(y);tr[y][0]=fa[x]=0;
}
void dfs(int now,int u)
{
for(int i=0;i<g[now].size();i++)
{
int v=g[now][i];
if(v==u) continue;
fa[v]=now;
dfs(v,now);
}
}
void update(int rt)
{
splay(rt);
col[rt]^=1;
pushup(rt);
}
void makeroot(int x)
{
access(x);splay(x);rev[x]^=1;
}
int query(int x)
{
makeroot(1);
access(x);splay(x);
if(sum[x]==0)return -1;
while(x)
{
if(sum[ tr[x][0] ]==0 && col[x]==1 ) return x;
else if(sum[tr[x][0]]) x=tr[x][0];
else x=tr[x][1];
}
}
void init()
{
for(int i=1;i<=n;i++) g[i].clear();
memset(col,0,sizeof(col));
memset(fa,0,sizeof(fa));
memset(sum,0,sizeof(sum));
memset(rev,0,sizeof(rev));
memset(tr,0,sizeof(tr));
}
int main()
{
while(scanf("%d%d",&n,&Q)!=EOF)
{
init();
int a,b;
for(int i=1;i<n;i++)
{
scanf("%d%d",&a,&b);
g[a].push_back(b);
g[b].push_back(a);
}
dfs(1,0);
while(Q--)
{
scanf("%d%d",&a,&b);
if(a==0) update(b);
else printf("%d\n",query(b));
}
}
return 0;
}