题意:约翰要修建光纤网络,给你一个矩阵,表示 i j 连接需要的花费,求出最小的花费
思路:求最小生成树,用kruskal算法直接算就行,在矩阵中把路给提出来就行了,借助了并查集,如果两个节点的根节点相同,那么会产生圈
#include <cstdio>
#include <iostream>
#include <vector>
#include <cmath>
#include <queue>
#include <algorithm>
#include <cstring>
const int maxn = 110;
const int INF = 1e6;
using namespace std;
struct edge{
int u, v, cost;
};
bool cmp(edge e1, edge e2){
return e1.cost < e2.cost;
}
int N;
int m;//边的数量
int mp[maxn][maxn];
edge es[maxn*maxn];
int par[maxn*maxn];
int ran[maxn*maxn];
//初始
void Init(int n){
for(int i = 0; i < n; i++){
par[i] = i;
ran[i] = 0;
}
}
//查询
int Find(int x){
if(x == par[x])
return par[x];
else
return par[x] = Find(par[x]);
}
//合并
void Unit(int x, int y){
x = Find(x);
y = Find(y);
if(x == y)
return ;
if(ran[x] > ran[y])
par[y] = x;
else{
par[x] = y;
if(ran[x] == ran[y])
ran[y]++;
}
}
//判断
bool Same(int x, int y){
return Find(x) == Find(y);
}
void Solve(){
long long ans = 0;
Init(N);
//排序,在不产生圈的情况下将最小边入队
sort(es, es+m, cmp);
for(int i = 0; i < m; i++)
if(!Same(es[i].u, es[i].v)){
Unit(es[i].u, es[i].v);
ans += es[i].cost;
}
cout << ans << endl;
return ;
}
//测试函数
int main(){
//freopen("D:\\钢铁程序员\\程序数据\\074光纤网络.txt", "r", stdin);
while(~scanf("%d",&N)){
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++)
scanf("%d", &mp[i][j]);
//将矩阵中的路提取出来
m = 0;
for(int i = 0; i < N; i++)
for(int j = 0; j < N; j++){
if(0 < mp[i][j] && mp[i][j] < INF){
es[m].u = i;
es[m].v = j;
es[m++].cost = mp[i][j];
}
}
Solve();
}
return 0;
}