第四专题 Problem E

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。<br>

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。<br><br>当N为0时输入结束。

 

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

 

Sample Input

31 2 1 01 3 2 02 3 4 031 2 1 01 3 2 02 3 4 131 2 1 01 3 2 12 3 4 10

 

Sample Output

310

 

思路：

先将n个村庄的建造公路的成本排序，最开始用的二维数组，发现排序的时候比较坑，然后改成的结构体数组，然后再卡一个数组，记录节点间是否连通，主要思想还是prim算法，最小生成树。

源码：

#include<iostream>#include<algorithm>using namespace std;struct node{   int a,b;   int len;}per[5000];int cmp(node x,node y){    if(x.len!=y.len) return x.len<y.len;}int main(){    int i,sum,n,m,num;    int flag[200];    while(scanf("%d",&n),n!=0)    {       m=(n*(n-1))/2;       for(i=0;i<m;i++)       {          scanf("%d%d%d",&per[i].a,&per[i].b,&per[i].len);       }       sort(per,per+m,cmp);       for(i=1;i<=n;i++) flag[i]=1;       flag[per[0].a]=0;       sum=0;       for(i=0;i<m;i++)       {          num=flag[per[i].a]+flag[per[i].b];          if(num==1)          {             sum+=per[i].len;             flag[per[i].a]=flag[per[i].b]=0;             i=0;          }       }       printf("%d\n",sum);    }    return 0;}