九度OJ-题目1017 还是畅通工程

题目描述：
某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
输入：
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
输出：
对每个测试用例，在1行里输出最小的公路总长度。

样例输入：
3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
样例输出：
3
5
来源：
2006年浙江大学计算机及软件工程研究生机试真题

用Prime算法（C语言）

#include
#include

#define N 100
#define MAX 10000

int prim(int path[N][N], int n){
    int sum= 0, min= MAX;//总路径， 目前最短
    int pa, pb;//最短端点
    int visited[N];
    int num;//走过的点数
    int i, j;
    memset(visited, 0, sizeof(visited));
    visited[1]= 1;//1点走过
    num= 1;
    while(num< n){
        min= MAX;
        for(i= 1; i<= n; i++){
            if(visited[i]== 1){
                for(j= 1; j<= n; j++){
                    if(visited[j]== 0 && path[i][j]> 0){
                        if(path[i][j]< min){
                            min= path[i][j];
                            pa= i;
                            pb= j;
                        }
                    }
                }

            }
        }
        visited[pb]= 1;
        sum+= path[pa][pb];
        num++;
    }
    return sum;
}

int main(){
    int n;//村庄数
    int path[N][N];//路径长度
    int a, b, c;//临时变量
    while(scanf("%d", &n)!=EOF && n!= 0){
        memset(path, 0, sizeof(path));
        for(int i= 0; i< n*(n-1)/2; i++){
            scanf("%d %d %d", &a, &b, &c);
            path[a][b]= path[b][a]= c;
        }
        printf("%d\n", prim(path, n));
    }
    return 0;
}