牛客挑战赛64-A.Connect Graph

链接：https://ac.nowcoder.com/acm/contest/42819/A
来源：牛客网

题目描述 
有一个 n 个点, m1 条边的无向图，现在要在这个图上按顺序添加 m2 条无向边，每新添加一条边需要花费 1 天时间，问每个点最早何时与 1 号点联通。
注：以未加“之后的 m2条边”时作为第 0 天，即第 0 天时图上就有 m1条边，加入第一条边的时间为第一天，若加入第 i 条边之后某个点与 1 号点联通，且在此之前该点与 1 号点不联通，则称该点与点 1 联通的时间为 i。
输入描述:
第一行一个整数 n(3≤n≤2×10e5)。

第二行两个整数 m1,m2(1≤m1+m2≤4×10e5)。

下面 m1 行，一行两个整数 u,v(1≤u,v≤n)，表示初始的一条边。

下面 m2行，一行两个整数 u,v(1≤u,v≤n)，表示新添加的一条边。

输出描述:
共 n 行，对于第 i 行输出一个整数表示点 i 与点 1 联通的时间，如果永远无法与点 1 联通，输出 -1。
 

示例1
输入
5
0 7
1 2
3 4
2 5
1 5
4 3
1 4
2 4
输出
0
1
6
6
3
说明

如图，其中边上的数字为加入这条边的时间。

该题每当新的不与1连通的一个点i与1连通时，除了要更新i与1连通的时间，还得更新与i连通的其他点的连通时间。(既然该点i之前不与1连通，那么与i连通的其他点之前也必然不与1连通。而i与1连通后，则与i连通的所有点都与1连通了，所以要更新)。
由于得一次性更新所有连通的点，所以得想办法把它们的状态绑在一起，因此可以用并查集去写。

用一个数组记录每个集合根节点与1连通的时间，当增加的一条边让点i与1连通时，直接让记录i的根节点的数组更新，这样所有与i连通的点也一起更新了。当增加一条边把两个都不与1连通的点连接时，直接让该两个点所在集合合并。

代码如下：

#include<iostream>
#include<cstring>
using namespace std;
const int N = 2e5+5;
int rt[N];
int val[N]; //记录与1连通的时间 
int Find(int x)
{
	int r = rt[x];
	if(r==x) return x;
	
	return rt[x] = Find(rt[x]);
}
void add(int x, int y)
{
	int rx = Find(x), ry = Find(y);
	if(rx!=ry)
	{
		if(rx==1) rt[ry] = rx;		//当其中一点的根节点是1时，让另外一个点连向1 
		else if(ry==1) rt[rx] = ry;
		else
			rt[rx] = ry; //如果两个点的根节点都不是1，那么随便连 
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n, m1, m2;
	memset(val, -1, sizeof(val)); //初始为-1 
	cin >> n >> m1 >> m2;
	for(int i=1; i<=n; i++) rt[i] = i; //初始化并查集数组 
	for(int i=0; i<m1; i++)
	{
		int x, y;
		cin >> x >> y;
		add(x, y);
	}
	val[1] = 0; //最初与1连通的点，连通时间为0 
	for(int i=1; i<=m2; i++)
	{
		int x, y;
		cin >> x >> y;
		int rx = Find(x), ry = Find(y);
		if(val[rx]>=0 && val[ry]>=0) continue; //如果两个根节点的val数组都不为-1，表示都已经与1连接。直接跳过，不用更新 
		else
		{
			if(val[rx]==-1 && val[ry]!=-1) val[rx] = i; //如果其中一个点已经和1连通，那么更新另外一个点的即可 
			else if(val[rx]!=-1 && val[ry]==-1) val[ry] = i;
			else add(x, y); //如果两个点都是不与1连通，那么直接合并 
		}
	}
	for(int i=1; i<=n; i++)
	{
		cout << val[Find(i)] << endl;
	}
	return 0;
}