luogu P1653 猴子

题目传送门：https://www.luogu.org/problemnew/show/P1653

题意：

有n只猴子，每只猴子最多抓住两只猴子。现在有m个时刻，每个时刻有一只猴子放下手中的一只猴子，求每只猴子的掉落时刻。注意：1号猴子永远不会掉落。

思路：

套路题，反向并查集。

每次放下猴子相当于反向重新连边，用个带权并查集维护即可。

代码：

#include<cstdio>
#include<algorithm>
using namespace std;
	int n,m;
	int a[200010][2],b[400010][2],fa[200010],ans[200010];
	bool flag[200010][2];
int find(int x)
{
	if(fa[x]==x) return x;
	int p=find(fa[x]);
	ans[x]=min(ans[x],ans[fa[x]]);
	return fa[x]=p;
}
void go(int x,int y,int z)
{
	int t1=find(x),t2=find(y);
	if(t1!=t2)
	{
		if(t1==1) fa[t2]=t1,ans[t2]=z;
		else fa[t1]=t2,ans[t1]=z;
	}
}
int main()
{
	scanf("%d %d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d %d",&a[i][0],&a[i][1]);
		fa[i]=i;
		ans[i]=m+1;
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d %d",&b[i][0],&b[i][1]);
		b[i][1]--;
		flag[b[i][0]][b[i][1]]=true;
	}
	for(int i=1;i<=n;i++)
	{
		if(a[i][0]!=-1&&!flag[i][0]) go(i,a[i][0],m);
		if(a[i][1]!=-1&&!flag[i][1]) go(i,a[i][1],m);
	}
	for(int i=1;i<=n;i++)
		ans[i]=m+1;
	for(int i=m;i>=1;i--)
		if(a[b[i][0]][b[i][1]]!=-1) go(b[i][0],a[b[i][0]][b[i][1]],i-1);
	for(int i=1;i<=n;i++)
		printf("%d\n",ans[i]>m?-1:ans[i]);
}