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

来源:互联网 发布:软考初级程序员视频 编辑:程序博客网 时间:2024/04/30 05:43

题目链接 :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;      }  } 
0 0