hdu1233 还是畅通工程(最小生成树 prim 算法)

来源:互联网 发布:抢购软件哪个好 编辑:程序博客网 时间:2024/04/30 03:18
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.

Source
浙大计算机研究生复试上机考试-2006年
#include<stdio.h>int n,map[105][105],S[105],node[105],sum,FN=1000000;void set_first(int n)//初始化{    for(int i=1;i<=n;i++)    {        node[i]=FN;        S[i]=0;  //0表示不在S集合中        for(int j=1;j<=n;j++)            map[i][j]=FN;    }}void prim(int stra){    int min,e;    sum=0;S[stra]=1;    for(int k=2;k<=n;k++)//k为S集合中加入了k个点,那么k个点在一棵树上    {        for(int i=1;i<=n;i++)//计算出与选择的点相连的点的权值        if(S[i]==0&&node[i]>map[stra][i])            node[i]=map[stra][i];        min=FN;        for(int i=1;i<=n;i++)//找出与加入S集合中的点相连的点的最小权值        if(min>node[i]&&S[i]==0)        {                min=node[i]; e=i;//记录位置        }        S[e]=1; sum+=min; stra=e;//sum为S集合中的点的权值    }}int main(){    int a,b,p;    while(scanf("%d",&n)>0&&n)    {        set_first(n);//初始化        for(int i=1;i<=n*(n-1)/2;i++)        {            scanf("%d%d%d",&a,&b,&p);            if(map[a][b]>p)//判断重边            {                map[a][b]=map[b][a]=p;            }        }        prim(1);//计算        printf("%d\n",sum);    }}

	
				
		
原创粉丝点击