prim算法经典应用 村庄与公路 hdu1863

畅通工程Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 8957    Accepted Submission(s): 3471

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

 

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

 

Sample Input

3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

 

Sample Output

3 ?

 

#include<stdio.h>
#include<string.h>
#define MAX 99999

#define LEN 101//这个要注意开大点

int dist[LEN];
int map[LEN][LEN];
int isvisited[LEN];

void init()
{
    int i,j;
    //题目只求1到100 ,LEN :101
    for(i=0;i<LEN;i++)
    {
        for(j=0;j<LEN;j++)
        {
            map[i][j]=MAX;
        }
    }
}

int prim(int n)
{
    int i,j,min,pos,sum;
    sum=0;
    memset(isvisited,0,sizeof(isvisited));
    for(i=1;i<=n;i++)
    {
        dist[i]=map[1][i];//初始化第一个点，将要到达的点距离
    }
    isvisited[1]=1;


    for(i=1;i<n;i++)
    {
        min=MAX;//不要放在外面
        for(j=1;j<=n;j++)
        {
            if(!isvisited[j] && dist[j]<min)//找出相邻将要到达的点(具有最小权值)
            {
                min=dist[j];
                pos=j;
            }
        }
        if(min==MAX)
            return -1;//无路可通
        sum+=min;
        isvisited[pos]=1;

        //加入pos点后，更新所有邻接点的权值，(如果有共同的)只交换较小的
        for(j=1;j<=n;j++)
        {
            if(!isvisited[j] && dist[j]>map[pos][j])//pos->j
            {
                dist[j]=map[pos][j];
            }
        }
    }
    return sum;
}

int main()
{

       int n,m;//n道路数(关系数)，m村庄数
       while(scanf("%d%d",&n,&m)!=EOF && n)
       {

              init();

              int i,a,b,w,ans;

              for(i=0;i<n;i++)
              {

                    scanf("%d%d%d",&a,&b,&w);

                     if(map[a][b]>w)
                     {

                            map[a][b]=map[b][a]=w;//对称矩阵
                     }

              }

              ans=prim(m);

              if(ans==-1){

                      puts("?");

              }

              else
              {
                  printf("%d\n",ans);
              }


       }
    return 0;
}