PAT 数据结构 06-图6. 公路村村通（30）Prim最小生成树算法

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

3 1
2 3 2

-1

5 4
1 2 1
2 3 2
3 1 3
4 5 4

-1

/*2015.7.16cyq*/
//Prim最小生成树算法
#include <iostream>
#include <vector>
#include <fstream>
#include <algorithm>
using namespace std;

const int MAX=2147483647;
//ifstream fin("case1.txt");
//#define cin fin

struct node{
int num;	//结点序号
int weight;	//能连通到已确定子图的最小边
int preNode;//最小边依附的结点,用于输出具体生成树
node():num(-1),weight(-1),preNode(-1){}
node(int n,int wei,int preN):num(n),weight(wei),preNode(preN){}
bool operator < (const node &a) const{
return weight<a.weight;
}
};

int main(){
int N,M;
cin>>N>>M;
vector<vector<int> > edges(N+1,vector<int>(N+1,-1));//-1表示不连通
int a,b,c;
while(M--){
cin>>a>>b>>c;
edges[a][b]=c;
edges[b][a]=c;
}

vector<node> AB;//已吸收的点,一开始只有结点1
AB.push_back(node(1,0,-1));//
vector<node> UD(N-1);//未吸收的点
for(int i=0;i<N-1;i++){
UD[i].num=i+2;//编号2到N
int tmp=edges[1][UD[i].num];
if(tmp>0){//能连到结点1的结点
UD[i].weight=tmp;
UD[i].preNode=1;
}else{//不能连到结点1的结点
UD[i].weight=MAX;
UD[i].preNode=-1;
}
}
int totalLength=0;
//vector<pair<int,int> > result;//用于输出具体生成树
while(!UD.empty()){
sort(UD.begin(),UD.end());
node cur=UD[0];//取出能以最小代价吸收的点
UD.erase(UD.begin());
if(cur.weight==MAX){
cout<<"-1"<<endl;
return 0;
}
totalLength+=cur.weight;
//	result.push_back(make_pair(cur.num,cur.preNode));
for(auto it=UD.begin();it!=UD.end();++it){//用该点修正未吸收的点
int w=edges[cur.num][(*it).num];
if(w>0){
if(w<(*it).weight){
(*it).weight=w;
(*it).preNode=cur.num;
}
}
}
}
cout<<totalLength<<endl;
//for(auto it=result.begin();it!=result.end();++it){//输出最小生成树
//	cout<<(*it).first<<" "<<(*it).second<<" "<<edges[(*it).first][(*it).second]<<endl;
//}
return 0;
}