#数据结构实验之图论六:村村通公路
###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