HDU 1233 又是畅通工程 [kruskal]+[prim]【最小生成树入门】

题目链接 :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): 37241 Accepted Submission(s): 16762

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

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <math.h>
#include <algorithm>
using namespace std;

int pre[1000];
int n,m,mintree;
struct node
{
    int a,b,l;
} c[10200];
void init(int n)
{
    for(int i=1; i<=n; i++)
        pre[i]=i;
    return ;
}
int findi(int x)
{
    int r=x;
    while(r!=pre[r])
        r=pre[r];

    int i=x,j;
    while(i!=j)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
int cmp(node a,node b)
{
    return a.l<b.l;
}

void kruskal()
{
    int fx,fy;
    for(int i=0; i<m; i++)
    {
        fx=findi(c[i].a),fy=findi(c[i].b);
        if(fx!=fy)
        {
            mintree+=c[i].l;
            pre[fy]=fx;
        }
    }
    return ;
}
int main()
{
    while(~scanf("%d",&n))
    {
        if(n==0) break;
        init(n);
        m=n*(n-1)/2;
        mintree=0;
        for(int i=0; i<m; i++)
        {
            scanf("%d%d%d",&c[i].a,&c[i].b,&c[i].l);
        }
        sort(c,c+m,cmp);
        kruskal();
        printf("%d\n",mintree);
    }
    return 0;
}

——————–

prim

#include <iostream> 
#include<stdio.h> 
#include<string.h> 
#include<math.h> 
const int N = 110;  
const int INF = 0x7ffffff;  

using namespace std;  

int v[N],d[N],mp[N][N],n;  

int prim()  
{  
    for(int i=1;i<=n;i++)    d[i]=INF,v[i]=0;  
    d[1]=0;  
    for(int i=0;i<n;i++)  
    {  
        int mmin=INF,t;  
        for(int j=1;j<=n;j++)  
            if(!v[j]&&mmin>d[j])  
                mmin=d[t=j];  
        v[t]=1;  
        for(int j=1;j<=n;j++)  
            if(!v[j])  
                d[j]=min(d[j],mp[t][j]);  
    }  
    int ans=0;  
    for(int i=1;i<=n;i++)    ans+=d[i];  
    return ans;  
}  

int main()  
{  
    while(~scanf("%d",&n)&&n)  
    {  
        for(int i=1;i<=n;i++)  
            for(int j=1;j<=n;j++)  
                mp[i][j]=INF;  
        int m=n*(n-1)/2;  
        for(int i=0;i<m;i++)  
        {  
            int u,v,t;  
            scanf("%d%d%d",&u,&v,&t);  
            if(mp[u][v]>t)  mp[u][v]=mp[v][u]=t;  
        }  
        cout<<prim()<<endl;  
    }  
}