题目传送门: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]); }