Agri-Net POJ - 1258

原题传送门

题意：约翰要修建光纤网络，给你一个矩阵，表示 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;
}