-
题目描述:
-
某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。
-
输入:
-
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 );随后的N(N-1)/2行对应村庄间的距离,每行给出一对正整数,分别是两个村庄的编号,以及此两村庄间的距离。为简单起见,村庄从1到N编号。
当N为0时,输入结束,该用例不被处理。
-
输出:
-
对每个测试用例,在1行里输出最小的公路总长度。
-
样例输入:
-
3 1 2 1 1 3 2 2 3 4 4 1 2 1 1 3 4 1 4 1 2 3 3 2 4 2 3 4 5 0
-
样例输出:
-
3
5
-
-
思路:求最小生成树,可用prim算法和krustral算法。
-
prim算法是:任取某一点,默认为1,找与之相邻的最短边,加入数组dist中,用sum记录最短边之和。
-
#include <iostream> #define INF 999999999 using namespace std; int map[105][105]; bool visit[105]; int dist[105]; int n; int prim() { for(int i=2;i<=n;++i) { visit[i]=0; dist[i]=map[i][1]; } visit[1]=1; int sum=0; for(int i=1;i<=n-1;++i)//从第一点开始遍历q3 { int temp=INF,pos; for(int j=1;j<=n;++j) { if(!visit[j]&&dist[j]<temp) { temp=dist[j]; pos=j; } } visit[pos]=1; sum+=dist[pos]; for(int j=1;j<=n;++j) { if(!visit[j]&&dist[j]>map[pos][j]&&map[pos][j]!=INF) dist[j]=map[pos][j]; } } return sum; } int main() { while(cin>>n) { if(n==0) break; int s,e,d; for(int i=1;i<=n;++i) dist[i]=INF; for(int i=1;i<=n*(n-1)/2;++i) { cin>>s>>e>>d; map[s][e]=map[e][s]=d;//初始化 } cout<<prim()<<endl; } return 0; }