poj---1287---Networking（最小生成树）

题目

思路

标准的模板题，最小生成树的板子直接打进去，我用的是Kruskal算法。

题解

#include <iostream>
#include <cstring>
#include <stdio.h>
#include <algorithm>
using namespace std; 
const int INF=0x3f3f3f3f;
const int MAXN=110;//最大点数
const int MAXM=100000;//最大边数
int F[MAXN];//并查集使用
int cost[MAXN][MAXN];
struct Edge{
	int u,v,w;
}edge[MAXM];//存储边的信息，包括起点/终点/权值
int tol=0;//边数，加边前赋值为 0
void addedge(int u,int v,int w){
	edge[tol].u=u;
	edge[tol].v=v;
	edge[tol++].w=w;
}
//排序函数，讲边按照权值从小到大排序
bool cmp(Edge a,Edge b){
	return a.w<b.w;
}
int find(int x){
	if(F[x]==-1)return x;
	else 
	{
		F[x]=find(F[x]);
		return F[x];
	}
}
//传入点数，返回最小生成树的权值，如果不连通返回 -1
int Kruskal(int n){
	memset(F,-1,sizeof(F));
	sort(edge,edge+tol,cmp);
	int cnt=0;//计算加入的边数
	int ans=0;
	for(int i=0;i<tol;i++){
		int u=edge[i].u;
		int v=edge[i].v;
		int w=edge[i].w;
		int t1=find(u);
		int t2=find(v);
		if(t1!=t2){
			ans+=w;
			F[t1]=t2;
			cnt++;
		}
		if(cnt==n-1)break;
	}
	if(cnt<n-1)return -1;//不连通
	else return ans;
}
int main()
{
	int n;
	bool pan=true;
	while(pan)
	{
		cin>>n;
		if(n==0)
		{
			break;
		}
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=n;j++)
			{
				cost[i][j]=INF;
				if(i==j)
				{
					cost[i][j]=0;
				}
			}
		}
		int t;
		cin>>t;
		for(int i=0;i<t;i++)
		{
			int a,b,c;
			cin>>a;
			cin>>b>>c;
			if(cost[a][b]>c)
			{
				cost[a][b]=c;
				cost[b][a]=c;
			}
		}
		tol=0;
		for(int i=1;i<=n;i++)
		{
			for(int j=1;j<=i;j++)
			{
				if(cost[i][j]!=0&&cost[i][j]!=INF)
				{
					addedge(i,j,cost[i][j]);
					addedge(j,i,cost[j][i]);
				}
			}
		}
		int ans=Kruskal(n);
		cout<<ans<<endl;
	} 
	return 0;
}