hdu 1233 还是畅通工程 （最小生成树）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1233

还是畅通工程

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

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

 

 

这是一个简单的最小生成树，第一次做，没有深刻的理解，自我感觉和dijkstra没有什么太大的本质区别，总之又是一种找最短路的很好的方法吧，值得参考~

 

 #include <iostream>
 #include <cstdio>
 using namespace std;
 int node[],map[][],n;
 const int INF=;

 void prim()
 {
     int vis[]= {};
     int sum=,Min,tm=,m;
     node[tm]=;
     vis[tm]=;//cout<<n<<endl;
     for (int k=; k<=n; k++)
     {

         Min=INF;
         for (int i=; i<=n; i++)
             if (!vis[i])
             {
                 if (node[i]>map[tm][i])
                     node[i]=map[tm][i];
                 if (Min>node[i])
                 {
                     Min=node[i];
                     m=i;
                 }
             }
         vis[m]=;
         sum+=Min;
         tm=m;
     }
     printf ("%d\n",sum);
 }

 int main ()
 {
     while (cin>>n&&n)
     {
         for(int i=; i<=n; i++)
         {
             node[i]=INF;
         }
         for (int i=; i<=n*(n-)/; i++)
         {
             int a,b,c;
             cin>>a>>b>>c;
                 map[a][b]=map[b][a]=c;
         }
         prim();
     }
     return ;
 }

第二种方法哈，也就是并查集+最小生成树了~~哇哈哈

详见代码。

 #include <iostream>
 #include <cstdio>
 #include <algorithm>
 using namespace std;
 int father[],sum;
 struct node
 {
     int s,d,f;
 } p[];

 bool cmp(const node &a,const node &b)
 {
     return a.f<b.f;
 }

 void set(int n)
 {
     for (int i=; i<=n; i++)
         father[i]=i;
 }

 int find(int a)
 {
     if (father[a]==a)
         return a;
     return father[a]=find(father[a]);
 }

 void Union(int x,int y,int z)
 {
     x=find(x);
     y=find(y);
     if (x!=y)
     {
         sum+=z;
         father[x]=y;
     }
 }

 int main ()
 {
     int n;
     while (cin>>n&&n)
     {
         sum=;
         set(n);
         for (int i=; i<=n*(n-)/; i++)
             cin>>p[i].s>>p[i].d>>p[i].f;
         sort(p+,p+n*(n-)/+,cmp);
         for (int i=; i<=n*(n-)/; i++)
         {
             Union(p[i].s,p[i].d,p[i].f);
         }
         printf ("%d\n",sum);
     }
     return ;
 }