题目链接在这里
题目大意
给出N个点,M条边(边有权值),在其中选边使得整个图联通,并且代价最小。
解题思路
最小生成树,莽就完事了。
代码如下
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <queue>
#include <functional>
#define clr(x) memset(x, 0, sizeof(x))
#define rep(i, n) for(int i = 0; i < n; ++i)
using namespace std;
const int MaxN = 60;
struct Node{
int x, y, val;
bool operator < (const Node& x) const {
return val > x.val;
}
};
int n, m;
int par[MaxN], r[MaxN];
priority_queue<Node> que;
void init(){
clr(r);
rep(i, MaxN) par[i] = i;
}
int Find(int x){
if(x == par[x]) return x;
return par[x] = Find(par[x]);
}
void unite(int x, int y){
x = Find(x);
y = Find(y);
if(x == y) return;
if(r[x] < r[y]) par[x] = y;
else{
par[y] = x;
if(r[x] == r[y]) ++r[x];
}
}
bool check(int x, int y){
return Find(x) == Find(y);
}
int main(){
ios::sync_with_stdio(false);
while(cin >> n && n){
while(!que.empty()) que.pop();
init();
cin >> m;
int x, y, val;
rep(i, m){
cin >> x >> y >> val;
que.push(Node{x, y, val});
}
int ans = 0;
while(--n){
Node node = que.top();
que.pop();
if(check(node.x, node.y)){
++n;
continue;
}
ans += node.val;
unite(node.x, node.y);
}
cout << ans << endl;
}
return 0;
}