hdu 1863 畅通工程(图论:最小生成树+并查集)

题目是很简单的最小生成树

但是有一个问题就是根据给出的数据能不能构成一个连通M个城市的最小生成树

我觉得用并查集来检查所有的城市是否归入一个集合即可

这样代码就很简单了,只需在Kruskal上加一个判断即可

0ms代码如下:

#include <cstdio>
#include <iostream>
#include <algorithm>
#define MAXN 10010
#define LL long long
using namespace std;

LL ans;
int m, n;
int u[MAXN], v[MAXN], w[MAXN];
int p[MAXN], r[MAXN];

int cmp(int i, int j) {
return w[i] < w[j];
}

int find(int x) {
return x==p[x] ? x : p[x] = find(p[x]);
}

bool Kruskal() {
int i, x, y, e, cnt;
ans = 0;
for(i=1; i<=m; ++i)//初始化从1开始,对应1-m个城市
p[i] = i;
for(i=0; i<n; ++i)
r[i] = i;
sort(r, r+n, cmp);
for(i=0; i<n; ++i) {
e = r[i];
x = find(u[e]);
y = find(v[e]);
if(x != y) {
ans += w[e];
p[x] = y;
}
}
cnt = 0;
for(i=1; i<=m; ++i) {//判断是否所有节点共用一个根
if(p[i] == i)
cnt++;
}
if(cnt == 1)
return true;
return false;
}

int main(void) {
int i;
while(cin >> n >> m, n) {
for(i=0; i<n; ++i) {
cin >> u[i] >> v[i] >> w[i];
}
if(Kruskal())
cout << ans << endl;
else cout << "?" << endl;
}
return 0;
}