HDU 1233 还是畅通工程 （自己种的第一棵最小生成树）

还是畅通工程

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


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

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

[align=left]Output[/align]
对每个测试用例，在1行里输出最小的公路总长度。

[align=left]Sample Input[/align]

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

[align=left]Sample Output[/align]

3
5

Hint
Hint

Huge input, scanf is recommended.

View Code

#include <set>
#include <map>
#include <list>
#include <cmath>
#include <ctime>
#include <deque>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cctype>
#include <cstring>
#include <sstream>
#include <fstream>
#include <cstdlib>
#include <cassert>
#include <iostream>
#include <algorithm>

using namespace std;
//Constant Declaration
/*--------------------------*/
//#define LL long long
#define LL __int64
const int M=110;
const int INF=1<<30;
const double EPS = 1e-11;
const double PI = acos(-1.0);
/*--------------------------*/
// some essential funtion
/*----------------------------------*/
void Swap(int &a,int &b){   int t=a;a=b;b=t; }
int Max(int a,int b){   return a>b?a:b;  }
int Min(int a,int b){   return a<b?a:b;  }
int Gcd(int a,int b){   while(b){b ^= a ^=b ^= a %= b;}  return a; }
/*----------------------------------*/
//for (i = 0; i < n; i++)
/*----------------------------------*/

int d[M];
int g[M][M];
int used[M];

void Init(int n)//初始化。
{
int i, j;
for (i = 1; i <= n; i++)
{
d[i] = INF;
}

for (i = 1; i <= n; i++)
{
for (j = 1; j <= n; j++)
{
g[i][j] = INF;
}
}

memset(used, 0, sizeof(used));
}

int Prim(int n)
{
d[1]  = 0;//默认d[1]为起点v1
int cnt, i, j;
int min, min_num;
int sum = 0;
for (cnt = 1; cnt <= n; cnt++)
{
min = INF;
min_num = 0;//防止非连通
for (i = 1; i <= n; i++)//生仔。
{
if (!used[i] && min > d[i])
{
min = d[i];
min_num = i;
}
}
if (min_num == 0)//非连通
{
return -1;
}

used[min_num] = 1;//别写成used[i]
sum += min;

for (i = 1; i <= n; i++)//松弛，更新。
{
if (!used[i] && d[i] > g[min_num][i])
{
d[i] = g[min_num][i];
}
}
}
return sum;
}

int main()
{
//freopen("in.txt","r",stdin);
//freopen("out.txt","w",stdout);
//int t, case1 = 0;
//scanf("%d", &t);
int n, m;
int i, j;
//scanf("%d%d", &n, &m);
while (scanf("%d", &n), n)
{
Init(n);
for (i = 0; i < n*(n-1)>>1; i++)
{
int a, b, c;
scanf("%d%d%d", &a, &b, &c);
if (g[a][b] > c)
{
g[a][b] = g[b][a] = c;
}

}
cout << Prim(n) << endl;
}

return 0;
}