hdu1233 还是畅通工程

还是畅通工程
Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 34251 Accepted Submission(s): 15470

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.

并查集+kruskal
数组开小了，一直提示TLE也是醉了。
注意用scanf读数据。

#include<iostream>#include<stdio.h>#include<cmath>#include<algorithm>#include<string>#include<cstring>#include<string.h>#include<map>#include<queue>#include<list>#include<stack>#include<cctype>#include<fstream>#include<sstream>#include<iomanip>#include<set>#include<vector>#include<cstdlib>#include<time.h>using namespace std;int p[105];struct node{    int a, b;    int val;}G[6000];bool cmp(node a, node b){    return a.val < b.val;}void make(){    for (int i = 0; i < 105; i++)        p[i] = i;}int find(int x){    if (x != p[x])        p[x] = find(p[x]);    return p[x];}void unio(int x, int y){    int px = find(x);    int py = find(y);    if (px == py)        return;    p[px] = py;}bool same(int x, int y){    return find(x) == find(y);}int kruskal(int m){    int s = 0;    for (int i = 0; i < m; i++)    {        if (!same(G[i].a, G[i].b))        {            unio(G[i].a, G[i].b);            s += G[i].val;        }    }    return s;}int main(){    int n;    while (scanf("%d",&n), n)    {        int m = n*(n - 1) / 2;        for (int i = 0; i < m; i++)        {            //cin >> G[i].a >> G[i].b >> G[i].val;            scanf("%d %d %d", &G[i].a, &G[i].b, &G[i].val);        }        sort(G, G + m, cmp);        make();        cout << kruskal(m) << endl;    }}