2018年9月24日提高组模拟赛 T3 观察

大意

给定一棵树，每个节点有黑白两种颜色，每次询问一个点与黑色点中的$LCA$中深度最大的那个$LCA$的深度

---

思路

$LCA$裸题

各种方法做，最稳的似乎是树链剖分。。。

本人用的是欧拉序+倍增，因为$LCA$中深度最高的两个点一定是在欧拉序中最接近的点，所以我们用一个$set$维护黑点的所有点，然后再在$set$中查找$dfn[x]$，这时得到了其对应的位置，让其分别与左右两点做一次$LCA$取深度更大的那个点即可

注意：本算法需要非常精妙的卡常技巧

---

代码

#include<set>
#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define r(i,a,b) for(register int i=(a);i<=(b);i++)
#define d(i,a,b) for(register int i=(b);i>=(a);i--)
#define N 800001
#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin)),p1==p2?EOF:*p1++)//fread优化
using namespace std;
char buf[1<<21],*p1,*p2;
bool ok;
inline int read() //读入优化
{
	int x=0,f=1; char c=getchar();
	while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}
	while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();
	return x*f;
}
inline void write(register int  x){if(x>9)write(x/10);putchar(x%10+48);return;}//输出优化
inline void writeln(register int x){write(x);putchar(10);return;}
int n,tot,cnt;
int dfn[N],id[N],f[N][24],dep[N],nxt[N],to[N],head[N],siz[N],q[N],m;
set <int> S;
set <int>::iterator it;
inline void add(register int x,register int y){nxt[++cnt]=head[x];to[cnt]=y;head[x]=cnt;}//建边
inline int lca(register int x,register int y) //倍增求LCA
{
	if(dep[x]<dep[y]) swap(x,y);
	int need=dep[x]-dep[y];
	for(int i=0;need;need>>=1,i++) if(need&1) x=f[x][i];
	if(x==y) return x;
	d(i,0,21) if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
	return f[x][0];
}
inline void bfs1() //求深度及子树大小
{
	int h=0,t=1;
	q[++h]=1;
	while(h<=t) 
	{
		int u=q[h++];
		siz[u]=1;
		for(register int i=head[u];i;i=nxt[i])
		{
			int v=to[i];
			dep[v]=dep[u]+1,f[v][0]=u;
			q[++t]=v;
		}
	}
	d(i,1,n) siz[f[q[i]][0]]+=siz[q[i]];
	return;
}
inline void bfs2() //求dfs序及时间戳，这两个bfs不可以放在一起打dfs，会被卡
{
	int h=0,t=1;
	q[++h]=1;dfn[1]=1; id[dfn[1]]=1;
	while(h<=t) 
	{
		int u=q[h++],last=u;
		for(register int i=head[u];i;i=nxt[i])
		{
			int v=to[i];
			dfn[v]=dfn[last]+(last!=u?siz[last]:1);
			id[dfn[v]]=v;
			q[++t]=v;
			last=v;
		}
	}
	return;
}
signed main() 
{
	n=read();m=read();
	r(i,1,n-1) {int u=read();add(u,i+1);}//输入+建边
	bfs1(),bfs2();
	for(register int j=1;(1<<j)<=n;j++) r(i,1,n) if(f[i][j-1]) f[i][j]=f[f[i][j-1]][j-1];//倍增思想
	dep[0]=-1;//默认0的深度为-1
	while(m--)
	{
		int x=read();
		if(x>0) 
		{
			if(!S.insert(dfn[x]).second) S.erase(dfn[x]);//若已经有过，删除，否则插入
			continue;
		}
		x=-x;
		if(S.empty()){putchar(48);putchar(10);continue;}
		int t1=0,t2=0;
		it=S.lower_bound(dfn[x]);//查找
		if(it!=S.end()) t1=lca(x,id[*it]);
		if(it!=S.begin()) it--,t2=lca(x,id[*it]);//计算左右两个的lca
		if(dep[t1]<dep[t2]) writeln(t2);else writeln(t1);//最后比较，输出
	}
}