数据结构实验之图论六:村村通公路
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
当前农村公路建设正如火如荼的展开,某乡镇政府决定实现村村通公路,工程师现有各个村落之间的原始道路统计数据表,表中列出了各村之间可以建设公路的若干条道路的成本,你的任务是根据给出的数据表,求使得每个村都有公路连通所需要的最低成本。
Input
连续多组数据输入,每组数据包括村落数目N(N <= 1000)和可供选择的道路数目M(M <= 3000),随后M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个村庄的编号和修建该道路的预算成本,村庄从1~N编号。
Output
输出使每个村庄都有公路连通所需要的最低成本,如果输入数据不能使所有村庄畅通,则输出-1,表示有些村庄之间没有路连通。
Sample Input
5 8
1 2 12
1 3 9
1 4 11
1 5 3
2 3 6
2 4 9
3 4 4
4 5 6
Sample Output
19
prim算法基本思想:
对于一个给定的图V,设置一个防护罩集合S来存放已被访问过的顶点,
然后执行n次以下步骤:
1,每次从V-S的集合中寻找与防护罩S最近的一个点u,访问该点u,将该点加入防护罩,
同时把这条最短边加入到最小生成树中。
2,如果能优化的话优化各顶点到当前最新防护罩的距离。具体做法为以当前的顶点u为参照点,看是否可以优化V-S集合中的点到防护罩的最小距离。
慎用memset,此题出错。
改用fill(d,d+MAXV,INF)
fill(bain[0],bian[0]+MAXV*MAXV,INF)
**/
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int MAXV=1001;
const int INF=1000000000;
int n;
int bian[MAXV][MAXV],d[MAXV],vis[MAXV];///分别记录权值,到防护罩的距离,访问位状态
int prim()
{
//memset(d,INF,sizeof(d));
fill(d,d+MAXV,INF);
d[1]=0; //先以点1为初始的防护罩集合
int ans=0;
for(int i=1;i<=n;i++)
{
int u=-1,min=INF;
for(int j=1;j<=n;j++)
{
if(vis[j]==0&&d[j]<min)
{
u=j;
min=d[j];
}
}
if(u==-1)
return -1;
vis[u]=1;
ans+=d[u];
///优化与u相连的未访问的点v到集合的最短距离
for(int v=1;v<=n;v++)
{
if(vis[v]==0&&bian[u][v]!=INF&&bian[u][v]<d[v])
{
d[v]=bian[u][v];
}
}
}
return ans;
}
int main()
{
int m,a,b,c;
while(cin>>n>>m)
{
fill(bian[0],bian[0]+MAXV*MAXV,INF);
fill(vis,vis+MAXV,0);
for(int i=0;i<m;i++)
{
cin>>a>>b>>c;
bian[a][b]=bian[b][a]=c;
}
int s=prim();
cout<<s<<endl;
}
return 0;
}