HDU 1223--やはりスムーズエンジニアリング【prim&&kruskal】

やはりスムーズな工事です
Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others) Total Submission(s): 27489    Accepted Submission(s): 12243
Problem Description
ある省は田舎の交通状況を調査し、得られた統計表には任意の2つの村間の距離がリストされている.省政府の「円滑な工事」の目標は、全省のどの2つの村の間でも道路交通を実現できるようにすることである(しかし、直接的な道路がつながっているとは限らず、間接的に道路を通過すればよい)、敷設された道路の総長さを最小限に抑えることを要求している.最小の道路の全長を計算してください.
 
Input
テスト入力には、いくつかのテスト例が含まれます.各試験例の1行目は、村の数N(<100)を与える.その後のN(N−1)/2行は村間の距離に対応し,各行には2つの村の番号とこの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年
 
ずっと前から書いていた問題ですが、当時はkruskalで書いていたので、面倒だと思ってprimeで書いたことがありませんが、先日長木さんが簡単なテンプレートを教えてくれて、試してみましたが、よく理解できました.

#include<stdio.h>
#include<string.h>
#define inf 0x3f3f3f3f

int vis[1010];
int dist[1010];
int map[100][100];
int n;

int getnext()
{
    int i,pos=-1,dis=inf;
    for(i=1;i<=n;++i)
        if(!vis[i]&&dist[i]<dis){
            dis=dist[i];pos=i;
        }
    return pos;
}

int prim()
{
    int sum=0,i,u,v;
    for(i=1;i<=n;++i){
        vis[i]=0;dist[i]=inf;
    }
    u=1;dist[1]=0;
    while(u!=-1){
        vis[u]=1;sum+=dist[u];
        for(i=1;i<=n;++i)
            if(!vis[i]&&dist[i]>map[u][i])
                dist[i]=map[u][i];
        u=getnext();

    }
    return sum;
}

int main ()
{
    int m,a,b,c,i;
    while(scanf("%d",&n),n){
        memset(map,0x3f3f3f,sizeof(map));
        memset(vis,0,sizeof(vis));
        m=n*(n-1)/2;
        for(i=0;i<m;++i){
            scanf("%d%d%d",&a,&b,&c);
            if(c<map[a][b])
                map[a][b]=map[b][a]=c;
        }
        printf("%d
",prim());
    }
    return 0;
}

次は前にkruskalで書いたものです

#include<stdio.h>
#include<algorithm>
using namespace std;
struct app{
    int st,ed,w;
};
int p[1000];
int find (int x)
{
    return p[x]==x?x:p[x]=find(p[x]);
}
int mergr(int a,int b)
{
    int x1,x2;
    x1=find(a);
    x2=find(b);
    if(x1!=x2)
    {
    p[x1]=x2;
    return 1;
    }
    return 0;
}
int cmp(app s1,app s2)
{
    return s1.w<s2.w;
}
int main ()
{
    int n,m,sum=0,con,i;
    app s[10000];
    while(scanf("%d",&n),n)
    {    
        con=0;
        m=n*(n-1)/2;
        for(i=1;i<=m;i++)
        scanf("%d%d%d",&s[i].st,&s[i].ed,&s[i].w);
        sort(s+1,s+1+m,cmp);
        for(i=1;i<=n;i++)
        p[i]=i;
        for(sum=0,i=1;i<=m;i++)
            if(mergr(s[i].st,s[i].ed))
            {
                        sum=sum+s[i].w;    
            }

        printf("%d
",sum);
    }
    return 0;
}