使用Kruskal算法的思想,按边权递增的顺序遍历所有的边,若遍历到的边的两个顶点分属于不同的集合,则将两个集合合并。集合用并查集来表示。
#include<iostream>
#include<algorithm>
using namespace std;
int Tree[1000];
int ans;
int findRoot(int x)
{
if (Tree[x] == -1)
{
return x;
}
else
{
int tmp = findRoot(Tree[x]);
Tree[x] = tmp;
return tmp;
}
}
struct Edge
{
int a, b;
int cost;
bool operator < (const Edge &A) const
{
return cost < A.cost;
}
}edge[1000];
int main()
{
int n;
cin >> n;
for (int i =1; i <= n*(n - 1) / 2; i++)
{
cin >> edge[i].a >> edge[i].b >> edge[i].cost;
}
sort(edge + 1, edge + 1 + n*(n - 1) / 2);
for (int i = 1; i <= n; i++)
{
Tree[i] = -1;
}
for (int i = 1; i <= n*(n - 1)/2; i++)
{
int a = findRoot(edge[i].a);
int b = findRoot(edge[i].b);
if (a != b)
{
Tree[a] = b; //合并两个集合
ans += edge[i].cost;
}
}
cout << ans << endl;
return 0;
}