3362-数据结构实验之图论六：村村通公路

#数据结构实验之图论六：村村通公路

###Problem Description

有n个城市，其中有些城市之间可以修建公路，修建不同的公路费用是不同的。现在我们想知道，最少花多少钱修公路可以将所有的城市连在一起，使在任意一城市出发，可以到达其他任意的城市。

###Input

输入包含多组数据，格式如下。
第一行包括两个整数n m，代表城市个数和可以修建的公路个数。(n <= 100， m <=10000)
剩下m行每行3个正整数a b c，代表城市a 和城市b之间可以修建一条公路，代价为c。

###Output

每组输出占一行，仅输出最小花费。
###Example Input
3 2
1 2 1
1 3 1
1 0
###Example Output
2
0

代码块

/*一、Prim算法（加点法）
用邻接矩阵存储
从村庄1开始，寻找和村庄1连通的有最小权值的村庄，之后再从该村庄找下一个有最小权值的村庄*/

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>   
#include<climits>
#define INF 2147483647
using namespace std;
int vis[10000];//标志数组，记录最小权值的村庄
int dis[1000][1000];//邻接矩阵
int g[10000];//记录其他村庄集合和已选村庄集合的最低成本

int prime(int dis[][1000],int n)
{
	int max,min,sum=0;
	int i,j;
	int p;
	memset(vis,0,sizeof(vis));//把标志数组置0 
	vis[1]=1;//假设数组从第一个数组开始 把1标记
	for(i=0;i<=n;i++)
	 g[i]=dis[1][i];
	for(i=1;i<n;i++)
	{
	   max=INF;
	    p=-1;
	    for(j=1;j<=n;j++)
	    {
	    	if(vis[j]==0 && max>g[j])
	    	{
	    	   max=g[j];
	    	    p=j;//记录下一个村庄代替当前村庄
	    	}
	    }
	    	if(max==INF)
	    	    return -1;//此图不连通
	    	sum=sum+max;
	    	vis[p]=1;//重新选取起始点
	    	for(j=1;j<=n;j++)//更新边
	    	{
	    		if(vis[j]==0 && g[j]>dis[p][j])
	    		{
	    			g[j]=dis[p][j];
				}
			}		
	}
	return sum;
}

int main()
{
	int n,m,u,v,w;
	int i,j;
	while(scanf("%d %d",&n,&m)!=EOF)
	{
		for(i=1;i<=n;i++)
		{
			for(j=1;j<=n;j++)
			{
				if(i==j)
				{
					dis[i][j]=0;
				}
				else
				{
					dis[i][j]=INF;
				}
			}
		}
		for(i=0;i<m;i++)
		{
			cin>>u>>v>>w;//依次输入村庄编号和成本
			if(dis[u][v]>w)
			{
				dis[u][v]=w;//无向图 预算成本赋值
				dis[v][u]=w;
			}
		}
		cout<<prime(dis,n)<<endl;	//调用prim
	}
	return 0;
}

/*二、kruskal算法（加边法）
按照权值的大小从小到大排序，从权值最小的边开始遍历每一条边，看两村庄是否连通，如果不连通，合并两个连通分量，直到所有的节点都在同一个连通分量中。*/
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>   
#include<climits>
#include<stdlib.h>
using namespace std;
int vis[1111];//标记各顶点所属的连通分量
int n,m,sum;
struct node
{
	int a,b,c;
}s[1111];

int f(int x)//找x是否在连通分量上 类似于并查集
{
	while(x!=vis[x])
	 x=vis[x];
	 return x;
}
void kruskal()//克鲁斯卡尔
{
	int p,q,i;
	for(i=0;i<m;i++)
	{
		p=f(s[i].a);
		q=f(s[i].b);
		while(p!=q)//表明所选的村庄是两个不同的连通分量
		{//p和q分别代表两个连通分量，如果村庄分别属于不同的连通分量，就合并两个连通分量
			sum=sum+s[i].c;
			vis[p]=q;
		}
	}
}

int cmp(const void *a,const void *b )
//按村庄之间的预算成本从小到大排序
{
	return (*(struct node *)a).c>(*(struct node *)b).c?1:-1;
}
int main()
{
	int i;
	int l;
	while(scanf("%d %d",&n,&m)!=EOF)
	{
		for(i=1;i<=n;i++)//初始化连通分量，当前的每一个城市单独作为连通分量
		 vis[i]=i;
		 sum=0;
		 for(i=0;i<m;i++)
           scanf("%d %d %d",&s[i].a,&s[i].b,&s[i].c);
		 qsort(s,m,sizeof(s[0]),cmp);
		 kruskal();
		 l=0;
		 for(i=1;i<=n;i++)
		   if(f(i)==i)
               l++;//只有符合法f（5）==5这个要求，l才能++
		   if(l!=1)
		     printf("-1\n");	
		   else
		     printf("%d\n",sum);	 
	}
	return 0;
}

###脚注

prim

vis 0 0 0 0 0

     1    2    3    4    5   
 1   0    12   9   11    *3*
 2   12    *0*   6    9   INF
 3   9     *6*   0    4   INF
 4  11     9   *4*    0    6
 5   3    INF INF   *6*    0

3+6+4+6=19
dis数组存储矩阵 vis 数组用来标记下标
g数组用来存储矩阵第一行数据，在村庄编号不断的变换下，g数组内的数据不断更新
g数组
0 12 9 11 3
0 12 9 6 3
0 9 4 6 3
0 6 4 6 3