原题链接:
http://soj.sysu.edu.cn/1083
思路:最小生成树。Prim算法求解。
代码:
//顶点编号为1-NumPoints
//数据输入和初始化应该根据具体情况而定
//Edges[Node2][Node1]=Edges[Node1][Node2]
#include <iostream>
using namespace std;
#define INF 2000000000
int Edges[105][105];
int Post[55];
bool Visited[55];
int NumPoints,NumRoads;
int find_min_node(){
int res=1,min=Post[1];
for(int i=1;i<=NumPoints;i++)
if(Post[i]<min){
min=Post[i];
res=i;
}
return res;
}
int Prim(){
int CostMinTree=0;
Visited[1]=true;
int add_node=1;
for(int i=2;i<=NumPoints;i++){
for(int j=1;j<=NumPoints;j++)
if((!Visited[j])&&(Edges[add_node][j]<Post[j]))
Post[j]=Edges[add_node][j];
add_node=find_min_node();
CostMinTree+=Post[add_node];
Visited[add_node]=true;
Post[add_node]=INF;
}
return CostMinTree;
}
int main(){
while(cin>>NumPoints){
if(NumPoints==0)
break;
cin>>NumRoads;
for(int i=1;i<=NumPoints;i++){
Visited[i]=false;
Post[i]=INF;
for(int j=1;j<=NumPoints;j++)
Edges[i][j]=INF;
}
for(int i=1;i<=NumRoads;i++){
int Node1,Node2,edge;
cin>>Node1>>Node2>>edge;
if(edge<Edges[Node1][Node2]){
Edges[Node1][Node2]=edge;
Edges[Node2][Node1]=edge;
}
}
cout<<Prim()<<endl;
}
return 0;
}