7-10 公路村村通(30 分)
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式:
输入数据包括城镇数目正整数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
作者: 陈越
单位: 浙江大学
时间限制: 400ms
内存限制: 64MB
#include<iostream>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
int Nv,Ne;
int Parent[1005];
struct Edge{
int c;//cost
int f;//from
int t;//to
bool operator <(const Edge &rhs) const{
return c>rhs.c;
}
};
priority_queue<Edge>Q;
vector<Edge>V;
void Read(){
int a,b,c;
for(int i=0;i<Ne;i++){
cin>>a>>b>>c;
Edge E;
E.c=c;
E.f=a;
E.t=b;
Q.push(E);
}
for(int i=1;i<Nv+1;i++)
Parent[i]=-1;
}
int Find(int x){
if(Parent[x]<0)
return x;
return Parent[x]=Find(Parent[x]);
}
void Union(int rt1,int rt2){
if(Parent[rt1]<Parent[rt2]){
Parent[rt1]=Parent[rt1]+Parent[rt2];
Parent[rt2]=rt1;
}else {
Parent[rt2]=Parent[rt2]+Parent[rt1];
Parent[rt1]=rt2;
}
}
bool Check(Edge E){
int rt1=Find(E.f);
int rt2=Find(E.t);
//cout<<"Test"<<E.f<<" "<<rt1<<" "<<endl;
if(rt1!=rt2){
Union(rt1,rt2);
return true;
}
else
return false;
}
void Kruskal(){
V.clear();
int cnt=0;
while((V.size()+1<Nv)&&Q.size()>0){
Edge E=Q.top();
Q.pop();
if(Check(E)==true){
V.push_back(E);
cnt+=E.c;
//cout<<V.size()<<endl;
}else
continue;
}
if(V.size()+1<Nv)
cout<<"-1"<<endl;
else
cout<<cnt<<endl;
}
int main(){
//freopen("in.txt","r",stdin);
cin>>Nv>>Ne;
Read();
Kruskal();
//fclose(stdin);
return 0;
}