中国大学MOOC-陈越、何钦铭-数据结构
08-图7 公路村村通 (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
算法思想:
最小堆存边结点,然后直接用kruskal建最小生成树的思想累计成本,用并查集判断结点间是否连通(一看题还以为要建图的,然后发现用个最小堆+kruskal直接ok)。
注意点:不能用Visited[ ]数组等于0和1判断结点时候已加入顶点集,如果用,问:v1和v2连通,v3和v4连通,那v2和v3怎么连通呢?(在这里挂了近两个小时,哭)
/***************2019.5.4-9:20-12:00 12:40-13:00***********/
//08-图7 公路村村通 3h
#include<cstdio>
#define MaxSize 1008
typedef struct ENode *Edge;
struct ENode{
int V1,V2;
int weight;
}E[MaxSize*3],tmp,X;
int Home[MaxSize];
void Insert(int i){
tmp=E[i];
for(i;E[i/2].weight>tmp.weight;i=i/2){
E[i]=E[i/2];
}
E[i]=tmp;
}
void Delete(int end){
if(end==0) return;
tmp=E[end];
int Parent,Child;
for(Parent=1;Parent*2<=end;Parent=Child){
Child=Parent*2;
if(Child!=end&&E[Child].weight>E[Child+1].weight) Child+=1;
if(tmp.weight<=E[Child].weight) break;
else E[Parent]=E[Child];
}
E[Parent]=tmp;
}
int Together(int V1,int V2){
int P1=V1;
int P2=V2;
while(Home[P1]>0) P1=Home[P1];
while(Home[P2]>0) P2=Home[P2];
if(P1==P2) return 1;
else {
if(Home[P1]<Home[P2]){
Home[P1]=Home[P1]+Home[P2];
Home[P2]=P1;
}
else{
Home[P2]=Home[P1]+Home[P2];
Home[P1]=P2;
}
return 0;
}
}
int main(){
int N,M;
int i;
scanf("%d%d",&N,&M);
E[0].weight=-1;
for(i=1;i<=M;i++){
scanf("%d%d%d",&E[i].V1,&E[i].V2,&E[i].weight);
Insert(i);
}
int end=M;
long long Cost=0;
int Enum=0;
for(i=1;i<=N;i++) Home[i]=-1;
while(1){
X=E[1];
Delete(end--);
if(end<0) break;
/* if(!Visited[X.V1]||!Visited[X.V2]){//这里不能保证v1和v2有边 v3和v4有边 然后连通这两个顶点
Visited[X.V1]=1;
Visited[X.V2]=1;
Cost+=X.weight;
Enum++;
if(Enum==N-1) break;
}*/
if(!Together(X.V1,X.V2)){//在一起跳过,不在一起在一起
Cost+=X.weight;
Enum++;
if(Enum==N-1) break;
}
}
if(Enum==N-1) printf("%lld\n",Cost);
else printf("-1\n");
return 0;
}