【BZOJ3319】黑白樹

Description

給定一棵樹，邊的顏色為黑或白，初始時全部為白色。維護兩個操作：
1.查詢u到根路徑上的第一條黑色邊的標號。
2.將u到v 路徑上的所有邊的顏色設為黑色。
Notice:這棵樹的根節點為1

Input

第一行兩個數n,m分別表示點數和操作數。
接下來n-? 1行，每行2個數u,v.表示一條u到v的邊。
接下來m行，每行為以下格式：
1 v 表示第一個操作
2 v u 表示第二種操作

Output

對於每個詢問，輸出相應答案。如果不存在，輸出0。

Sample Input

Sample Output

HINT

對於 100% 的數據：n，m<=10^6

題解：本題要用到兩邊並查集。先用並查集預處理出每條邊第一次變黑的時間，然後時間倒流。如果這個點是白點，則將該點的並查集與其父親的並查集合並；如果是黑點則不合並。這樣，每個點所在的並查集的根節點的邊就是路徑上第一個黑邊。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
const int maxn=1000010;
int n,m,cnt;
int to[maxn<<1],next[maxn<<1],head[maxn],vis[maxn],f[maxn],v[maxn];
int dep[maxn],fa[maxn],son[maxn],top[maxn],siz[maxn],q[maxn],ans[maxn];
inline int rd()
{
	int ret=0,f=1;	char gc=getchar();
	while(gc<‘0‘||gc>‘9‘)	{if(gc==‘-‘)f=-f;	gc=getchar();}
	while(gc>=‘0‘&&gc<=‘9‘)	ret=ret*10+gc-‘0‘,gc=getchar();
	return ret*f;
}
void add(int a,int b)
{
	to[++cnt]=b,next[cnt]=head[a],head[a]=cnt;
}
void dfs1(int x)
{
	siz[x]=1;
	for(int i=head[x];i;i=next[i])	if(to[i]!=fa[x])
	{
		fa[to[i]]=x,dep[to[i]]=dep[x]+1,v[to[i]]=(i+1)>>1,dfs1(to[i]),siz[x]+=siz[to[i]];
		if(siz[to[i]]>siz[son[x]])	son[x]=to[i];
	}
}
void dfs2(int x,int tp)
{
	top[x]=tp;
	if(son[x])	dfs2(son[x],tp);
	for(int i=head[x];i;i=next[i])	if(to[i]!=fa[x]&&to[i]!=son[x])	dfs2(to[i],to[i]);
}
int lca(int x,int y)
{
	while(top[x]!=top[y])
	{
		if(dep[top[x]]<dep[top[y]])	swap(x,y);
		x=fa[top[x]];
	}
	if(dep[x]<dep[y])	return x;
	return y;
}
int find(int x)
{
	return (f[x]==x)?x:(f[x]=find(f[x]));
}
int main()
{
	n=rd(),m=rd();
	int i,j,a,b,c;
	for(i=1;i<n;i++)	a=rd(),b=rd(),add(a,b),add(b,a);
	for(i=1;i<=n;i++)	f[i]=i;
	dep[1]=1,dfs1(1),dfs2(1,1);
	memset(head,0,sizeof(head)),cnt=0;
	for(i=1;i<=m;i++)
	{
		if(rd()==1)	q[i]=rd();
		else
		{
			a=rd(),b=rd(),c=lca(a,b);
			a=find(a),b=find(b);
			while(dep[a]>dep[c])	f[a]=find(fa[a]),add(i,a),vis[a]=1,a=f[a];
			while(dep[b]>dep[c])	f[b]=find(fa[b]),add(i,b),vis[b]=1,b=f[b];
		}
	}
	for(i=1;i<=n;i++)	f[i]=!vis[i]?fa[i]:i;
	for(i=m;i>=1;i--)
	{
		if(q[i])	ans[i]=v[find(q[i])];
		else	for(j=head[i];j;j=next[j])	f[to[j]]=find(fa[to[j]]);
	}
	for(i=1;i<=m;i++)	if(q[i])	printf("%d\n",ans[i]);
	return 0;
}

【BZOJ3319】黑白樹 並查集