HDU---1233-畅通工程（最小生成树prim）

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 30189    Accepted Submission(s): 13487

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

31 2 11 3 22 3 441 2 11 3 41 4 12 3 32 4 23 4 50

 

Sample Output

35
Hint
Hint
 Huge input, scanf is recommended.

#include <iostream>#include <cstdio>#include <cmath>#include <algorithm>#include <cstdlib>#include <cstring>#include <limits.h>using namespace std;#define MIN INT_MAX#define MAX_Point 120#define MAX_Edge 14400int flag1=0;double sum;double arr_list[MAX_Point][MAX_Point];struct Edge{    int point;    double lowcost;    int flag;} edge[MAX_Edge];double prim(int n){    int i,j=1,k=1,flag;    double min,sum2=0;    for(i=1; i<=n; i++ )    {        if(i!=j)        {            edge[i].point=i;            edge[i].lowcost=arr_list[j][i];            edge[i].flag=0;        }    }    edge[j].lowcost=0;    edge[j].flag=1;    for(i=2; i<=n; i++)    {        k=1;        min=MIN;        flag=0;        for(j=2; j<=n; j++)        {            if(edge[j].flag==0 && edge[j].lowcost<min)            {                k=j;                min=edge[j].lowcost;                flag=1;            }        }        if(!flag)return -1;        sum2+=min;        edge[k].flag=1;        for(j=2; j<=n; j++)        {            if(edge[j].flag==0 && arr_list[k][j]<edge[j].lowcost)            {                edge[j].point=k;                edge[j].lowcost=arr_list[k][j];            }        }    }    return sum2;}int main(){    double time_data;    int m;    int a,b;    while(cin>>m)    {        if(m==0)break;        sum=0;        for(int i=1; i<=m; i++)            for(int j=1; j<=m; j++)                arr_list[i][j]=INT_MAX;        for(int i=1; i<=m*(m-1)/2; i++)        {            cin>>a>>b>>time_data;            if(time_data<arr_list[a][b])            {                arr_list[a][b]=time_data;                arr_list[b][a]=time_data;            }        }        sum=prim(m);        printf("%.lf\n",sum);    }    return 0;}