并查集

发表于   |   分类于   |  

并查集

1

1
2
3
4
5
int findRoot(int x)
{
	if(tree[x]==-1) return x;
	else return findRoot(tree[x]);
}

非递归版本

1
2
3
4
5
6
7
8
9
int findRoot(int x)
{
	while(tree[x]!=-1)
	{
		x=tree[x];
	}
	int root=x;
	return root;
}

路径压缩

2

1
2
3
4
5
6
7
8
9
10
int findRoot(int x)
{
	if(next[i]==-1) return x;
	else
	{
		int temp=findRoot(tree[x]);
		tree[x]=temp;
		return temp;
	}
}

非递归版本

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
int findRoot(int x)
{
	int temp=x; //暂存x
	while(tree[x]!=-1)// 第一步找出root
	{
		x=tree[x];
	}
	int root=x;
	x=temp;
	while(tree[x]!=-1)//第二步，从x到root路径上所有的点都指向root
	{
		int t=tree[x];//暂存x的father
		tree[x]=root;
		x=t;
	}
	return root;
}

例子：畅通工程

描述

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。
省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，
只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？

input

测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。

output

对每个测试用例，在1行里输出最少还需要建设的道路数目。

sample input

4 2
1 3
4 3
3 3
1 2
1 3
2 3
5 2
1 2
3 5
999 0
0

sample output

1
0
2
998

杭电1232题

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
#include<stdio.h>
int tree[1000];

/*int findRoot(int x)
{
	if(tree[x]==-1) return x;
	else
	{
		int temp=findRoot(tree[x]);
		tree[x]=temp;
		return temp;
	}
}*/

int findRoot(int x)
{
	int temp=x;
	while(tree[x]!=-1)
	{
		x=tree[x];
	}
	int root=x;
	x=temp;
	while(tree[x]!=-1)
	{
		int t=tree[x];
		tree[x]=root;
		x=t;
	}
	return root;
}

int main()
{
	int N,M;
	while(scanf("%d",&N)!=EOF&&N!=0)
	{
		scanf("%d",&M);
		for(int i=1;i<=N;i++)
		{
			tree[i]=-1;
		}
		for(int i=1;i<=M;i++)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			int aRoot=findRoot(a);
			int bRoot=findRoot(b);
			if(aRoot!=bRoot)
			{
				tree[aRoot]=bRoot;
			}
		}
		int count=0;
		for(int i=1;i<=N;i++)
		{
			if(tree[i]==-1)
			{
				count++;
			}
		}
		printf("%d\n",count-1);
	}
	return 0;
}