还是畅通工程(Prim)

还是畅通工程

Time Limit : 4000/2000ms (Java/Other) Memory Limit : 65536/32768K (Java/Other)
Total Submission(s) : 66 Accepted Submission(s) : 53
Font: Times New Roman | Verdana | Georgia
Font Size: ← →
Problem Description
某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。
Input
测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。
Output
对每个测试用例，在1行里输出最小的公路总长度。
Sample Input
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
Sample Output
3
5
Hint
Hint
Huge input, scanf is recommended.
Source
浙大计算机研究生复试上机考试-2006年

这里用的是时间复杂度为n*n的算法，更高效最小堆算法还在学习中。

AC代码：

#include<iostream>#include<cstdio>#include<string>#include<algorithm>#include<cstring>#include<vector>#include<vector>#include<functional>#include<stack>#include<queue>#include<list>using namespace std;#define INF 9999999int vis[10000];int map[105][105];int v[100];int n;void init(){    memset(vis, 0, sizeof(vis));    for (int i = 0;i <= 104;i++)    {        for (int j = 0;j <= 104;j++)        {            map[i][j] = INF;        }    }}int prim(){    int temp, mint;int sum = 0;    vis[1] = 1;    for (int i = 1;i <= n;i++)    {        v[i] = map[1][i];    }    for (int i = 1;i < n;i++)    {        //cout << sum << endl;        mint = INF;temp = 0;        for (int j = 1;j <= n;j++)        {            if (!vis[j] && mint > v[j])            {                temp = j;                mint = v[j];            }        }        vis[temp] = 1;        sum += mint;        for (int j = 1;j <= n;j++)        {            if (map[temp][j] < v[j] && !vis[j])            {                v[j] = map[temp][j];            }        }    }    return sum;}int main(){    int x, y, t;    while (cin >> n&&n)    {        init();        for (int i = 1;i <= n*(n - 1) / 2;i++)        {            cin >> x >> y >> t;            if (t < map[x][y])            {                map[x][y] = map[y][x] = t;            }        }        cout << prim() << endl;    }    return 0;}