HDU1233（Kruskal&Prim两解）

https://vjudge.net/problem/HDU-1233

还是畅通工程

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

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

Huge input, scanf is recommended.

//Kruskal

 #include<iostream>
 #include<stdio.h>
 #include<string.h>
 #include<algorithm>
 #include<cmath>
 #include<vector>
 #include<queue>
 const int maxn=;
 #define ms(x,n) memset(x,n,sizeof x);
 const int inf=0x3f3f3f3f;
 using namespace std;
 int n,m;
 int u,v,w,parent[maxn];//ranks[maxn];
 struct node
 {
     int u,v,w;
     node(){u=,v=,w=inf;}
     //node(int uu,int vv,int ww){u=uu,v=vv,w=ww;}
 }edge[maxn*maxn];
 bool cmp(node a,node b)
 {
     return a.w<b.w;
 }
 int init(int n)
 {
     for(int i=;i<=n;i++)
         parent[i]=-;//,ranks[i]=-1;
 }
 int finds(int x)
 {
     if(parent[x]<) return x;
         return parent[x]=finds(parent[x]);
 }
 void unions(int x,int y)
 {
     x=finds(x),y=finds(y);
     if(x==y) return;
     if(parent[x]<parent[y])
         {
         parent[x]+=parent[y];
         parent[y]=x;
         }
     else
     {
         parent[y]+=parent[x];
         parent[x]=y;
     }
 }
 int kruskal()
 {
     sort(edge+,edge+m+,cmp);
     init(n);
     int ans=;
     for(int i=;i<=m;i++)
     {
         node a=edge[i];
         if(finds(a.u)!=finds(a.v))
         {
             unions(a.u,a.v);
             ans+=a.w;
         }
     }
     return ans;
 }
 int main()
 {
     ios::sync_with_stdio(false);
     cin.tie();
     while(~scanf("%d",&n),n)
     {
         m=n*(n-)/;
         for(int i=;i<=m;i++)
             edge[i]=node();
         for(int i=;i<=m;i++)
         {
             scanf("%d%d%d",&edge[i].u,&edge[i].v,&w);
             edge[i].w=min(edge[i].w,w);
         }
         printf("%d\n",kruskal());
     }
     return ;
 }

//Prim

 #include<iostream>
 #include<stdio.h>
 #include<string.h>
 #include<algorithm>
 #include<cmath>
 #include<vector>
 #include<queue>
 #define ms(x,n) memset(x,n,sizeof x);
 const int maxn=;
 const int inf=0x3f3f3f3f;
 using namespace std;
 int d[maxn],g[maxn][maxn];
 int n;
 bool vis[maxn];
 void prim()
 {
     int i,j,k,tmp,ans=;
     ms(d,0x3f);
     ms(vis,);
     d[]=;
     for(i=;i<=n;i++)
     {
         tmp=inf;
         for(j=;j<=n;j++)
         {
             if(!vis[j]&&tmp>d[j])
             {
                 tmp=d[j];
                 k=j;
             }
         }
         vis[k]=;
         ans+=tmp;
         for(j=;j<=n;j++)
         {
             if(!vis[j]&&d[j]>g[k][j])
                 d[j]=g[k][j];
         }
     }
     cout<<ans<<endl;
 }
 int main()
 {
     while(~scanf("%d",&n),n)
     {
         int m=n*(n-)/;
         for(int i=;i<=n;i++)
             for(int j=;j<=n;j++)
             if(i!=j)
             g[i][j]=inf;
         else
             g[i][j]=;
         for(int i=;i<=m;i++)
         {
             int u,v,w;
             scanf("%d%d%d",&u,&v,&w);
             g[u][v]=g[v][u]=w;
         }
         prim();
     }
     return ;
 }