杭电OJ 1162（C++）

计算最小生成树的总长度，使用 Kruskal算法。

#include <iostream>
#include <algorithm>
#include <iomanip>
using namespace std;
const int MAXN = 105;

int n; //点的数量
int m; //线的数量
int fa[MAXN]; //父结点数组，fa[i]表示第i个点的父结点

struct Point //点
{
	double x;
	double y;
}p[MAXN];

struct Line
{
	int s; //起点序号
	int e; //终点序号
	double length; //长度
}l[MAXN * MAXN];

bool cmp(Line l1, Line l2)
{
	return l1.length < l2.length;
}

//查找x的根结点
int root(int x)
{
	if (x != fa[x])
		return root(fa[x]);
	else
		return x;
}

double Kruskal()
{
	for (int i = 0; i < n; i++)
	{
		fa[i] = i;
	}
	double total = 0; //总长度
	for (int i = 0; i < m; i++)
	{
		int rootS = root(l[i].s); //起点的根结点
		int rootE = root(l[i].e); //终点的根结点
		if (rootS != rootE)
		{
			total += l[i].length;
			fa[rootE] = rootS;
		}
	}
	return total;
}

int main()
{
	while (cin >> n)
	{
		for (int i = 0; i < n; i++)
		{
			cin >> p[i].x >> p[i].y;
		}
		m = 0;
		for (int i = 0; i < n - 1; i++)
		{
			for (int j = i + 1; j < n; j++)
			{
				l[m].s = i;
				l[m].e = j;
				l[m].length = sqrt(pow((p[i].x - p[j].x), 2) + pow((p[i].y - p[j].y), 2));
				++m;
			}
		}
		sort(l, l + m, cmp); //根据线的长度从小到大排序
		cout << fixed << setprecision(2) << Kruskal() << endl;
	}
	return 0;
}

继续加油。