随着小Hi拥有城市数目的增加,在之间所使用的Prim算法已经无法继续使用了——但是幸运的是,经过计算机的分析,小Hi已经筛选出了一些比较适合建造道路的路线,这个数量并没有特别的大。
所以问题变成了——小Hi现在手上拥有N座城市,且已知其中一些城市间建造道路的费用,小Hi希望知道,最少花费多少就可以使得任意两座城市都可以通过所建造的道路互相到达(假设有A、B、C三座城市,只需要在AB之间和BC之间建造道路,那么AC之间也是可以通过这两条道路连通的)。
输入
每个测试点(输入文件)有且仅有一组测试数据。
在一组测试数据中:
第1行为2个整数N、M,表示小Hi拥有的城市数量和小Hi筛选出路线的条数。
接下来的M行,每行描述一条路线,其中第i行为3个整数N1_i, N2_i, V_i,分别表示这条路线的两个端点和在这条路线上建造道路的费用。
对于100%的数据,满足N<=10^5, M<=10^6,于任意i满足1<=N1_i, N2_i<=N, N1_i≠N2_i, 1<=V_i<=10^3.
对于100%的数据,满足一定存在一种方案,使得任意两座城市都可以互相到达。
输出
对于每组测试数据,输出1个整数Ans,表示为了使任意两座城市都可以通过所建造的道路互相到达至少需要的建造费用。
#include<iostream>
#include<algorithm>
#include<queue>
#include<vector>
#include<cstdio>
#include<cstring>
#include<cstdlib>
using namespace std;
#define MAX_POINT 100005
#define MAX_EDGE 1000005
struct Edge {
int edge_a;
int edge_b;
int len;
};
bool cmp (const Edge &lhs, const Edge &rhs) {
return lhs.len < rhs.len;
}
int N, M;
int p[MAX_POINT];
Edge e[MAX_EDGE];
int find (int x) {
if (x == p[x])
return x;
else
p[x] = find(p[x]);
return p[x];
}
void merge (int a, int b) {
p[find(a)] = find(b);
}
void GetGraph() {
for (int i = 0; i < M; ++i) {
cin >> e[i].edge_a >> e[i].edge_b >> e[i].len;
}
}
int Kruscal () {
int result = 0;
sort(e, e+M, cmp);
for (int i = 0; i < M; ++i) {
if (find(e[i].edge_a) == find(e[i].edge_b))
continue;
result += e[i].len;
merge(e[i].edge_a, e[i].edge_b);
}
return result;
}
int main (int args, char* argv[])
{
//freopen("F:/in.txt","r",stdin);
cin >> N >>M;
for (int i = 1; i <= N; i++)
p[i] = i;
GetGraph();
cout << Kruscal() << endl;
//fclose(stdin);
}