kruskal(最小生成树)
题目描述
第1行给出评估的道路条数 N、村庄数目M ( < 100 );
随后的N行,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间道路的成本(也是正整数)。为简单起见,村庄从1到M编号。
使任何两个村庄间都可以实现公路交通,但不一定有直接的公路相连,只要能间接通过公路可达即可,输出全省畅通需要的最低成本。若统计数据不足以保证畅通,则输出“?”。
const int maxn = 200;//N的上限
int father[maxn];//father[i]:表示i的父亲节点
int ran[maxn];//ran[i]:i的深度
int m, n;
struct node {//从from到to城市需要的费用为val
int from, to, val;
}s[maxn];
bool cmd(node a, node b) {//排序依据
return a.val < b.val;
}
void reset() {//重置、初始化
for (int i = 0; i < m; i++) {
father[i] = i;
}
memset(ran, 0, sizeof(ran));
}
int fin(int x) {//返回连接x的第一个节点
return father[x] = father[x] == x ? x : fin(father[x]);
}
void meg(int x, int y) {//将x和y连接起来
int a = fin(x);
int b = fin(y);
if (ran[a] < ran[b]) {
father[a] = b;
}
else {
father[b] = a;
if (ran[a] == ran[b]) {
ran[a]++;
}
}
}
bool same(int x, int y) {//x和y如果已经连接,返回ture
return fin(x) == fin(y);
}
int kruskal() {//返回全省畅通需要的最低成本
sort(s, s + n, cmd);
int sum = 0;
for (int i = 0; i < n; i++) {
if (same(s[i].from, s[i].to)) {
continue;
}
else {
sum += s[i].val;
meg(s[i].from, s[i].to);
}
}
return sum;
}