现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数N(≤1000)和候选道路数目M(≤3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出−1,表示需要建设更多公路。
输入样例:
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
输出样例:
12
注意点:prim算法可以看做是由dijkstra算法稍微改动一下就能得到,dijkstra是把每个点求出与源点的最短距离,而prim是把每个点添加到最小生成树里面就可以了,所有改动两行便可,详见代码。
#include <iostream>
#include <algorithm>
#include <cstdio>
#include <vector>
#include <stack>
#include <queue>
#include <cmath>
using namespace std;
const int maxn=1001;
const int inf=999999;
int n,m;
int dis[maxn],e[maxn][maxn];
bool visit[maxn];
int mincost=inf;
void prim(){
int cnt=0,cost=0;
dis[1]=0;
for(int i=1;i<=n;i++){
int u=-1,minn=inf;
for(int j=1;j<=n;j++){
if(visit[j]==false&&dis[j]<minn){
u=j;
minn=dis[j];
}
}
if(u==-1) break;
visit[u]=true;
cost+=dis[u];
cnt++;
for(int v=1;v<=n;v++){
if(visit[v]==false&&e[u][v]!=inf&&e[u][v]<dis[v])//和传统dijkstra算法的区别在于这里,直接添加到生成树里,而不是源点
dis[v]=e[u][v];
}
}
if(cnt==n) cout<<cost<<endl;
else cout<<"-1\n";
}
int main()
{
fill(dis,dis+maxn,inf);
fill(e[0],e[0]+maxn*maxn,inf);
cin>>n>>m;
for(int i=0;i<m;i++){
int a,b,c;
cin>>a>>b>>c;
e[a][b]=e[b][a]=c;
}
prim();
return 0;
}