HOJ 1879 继续畅通工程(最小生成树和并查集)

畅通工程。。。。。终于畅通了。。。。。
本题把给出已建道路的村庄集合先合并,后面再用Prime结合并查集直接暴力就可以了。

#include<iostream>
#include<cstring>
using namespace std;
const int NN = 100 + 5, INF = 99999999;
int a[NN][NN], f[NN];

int find(int x)
{//找到根节点
	if (x == f[x]) return x;
	f[x] = find(f[x]);
	return f[x];
}

int mer(int x, int y)
{
	x = find(x);
	y = find(y);
	if (x != y)
		f[x] = y;
	return 1;
}

int main()
{
	int n, x, y, z, k, sum;
	while (cin >> n && n)
	{
		sum = 0;
		for (int i = 0; i < n; i++)
		{
			f[i] = i;
			a[i][i] = INF;
		}
		for (int i = 0; i < n * (n - 1) / 2; i++)
		{
			scanf("%d %d %d %d", &x, &y, &z, &k);
			a[x - 1][y - 1] = z;
			a[y - 1][x - 1] = z;
			if (k) mer(x - 1, y - 1);//k为1表示两个地方之间已有道路,故两个集合合并
		}
		while (1)
		{//直接使用Prime算法即可
			int min0 = INF, flag = 1;
			for (int i = 0; i < n; i++)
				for (int j = 0; j <= i; j++)
					if (find(i) != find(j) && a[i][j] < min0)
					{
						min0 = a[i][j];
						x = i;
						y = j;
						flag = 0;
					}
			if (flag) break;
			mer(x, y);
			sum += a[x][y];
		}
		cout << sum << endl;
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_44724908/article/details/104183842