[MST]HDU 1233还是畅通工程

传送门:还是畅通工程

还是畅通工程

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

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.
  
  

 

Source

浙大计算机研究生复试上机考试-2006年

解题报告：

此题是最小生成树问题。此题是无向连通图。代码如下：

方法一：prim 邻接矩阵

#include<iostream>
#include<cstdio>
#include<cstring>
#define  INF   0x7fffffff
using namespace std;
int a[111][111],dist[111],n;
bool vis[111];
int main(){
    while(scanf("%d",&n)==1&&n){
        int aa,b,c;
        memset(vis,0,sizeof(vis));
        memset(a,0,sizeof(a));
        for(int i=1;i<=(n*(n-1))/2;i++){
            scanf("%d%d%d",&aa,&b,&c);
            a[aa][b]=c;
            a[b][aa]=c;
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
                if(a[i][j]==0)
                    a[i][j]=INF;
        for(int i=1;i<=n;i++)
            dist[i]=a[1][i];
        vis[1]=true;
        int num=1,sum=0;
        while(num<n){
            int res=INF,pos;
            for(int i=1;i<=n;i++)
                if(dist[i]<res&&!vis[i]){
                    pos=i;res=dist[i];
                }
            vis[pos]=true;
            sum+=res;
            for(int i=1;i<=n;i++)
                if(!vis[i])
                    dist[i]=min(dist[i],a[pos][i]);
            num++;
        }
        printf("%d\n",sum);
    }
    return 0;
}

方法二：prim 邻接表

#include<iostream>
#include<cstdio>
#include<cstring>
#define  INF   0x7fffffff
using namespace std;
struct edge{
    int v,next,cost;
}e[111*111];
int ecnt,first[111];
int n,v;
void addedge(int u,int v,int cost){
    e[++ecnt].next=first[u];
    e[ecnt].v=v;
    e[ecnt].cost=cost;
    first[u]=ecnt;
}
int prim(){
    bool vis[111];
    int dist[111],num=1,sum=0;
    memset(vis,0,sizeof(vis));
    vis[1]=true;
    for(int i=1;i<=n;i++)
        dist[i]=INF;
    for(int i=first[1];i!=-1;i=e[i].next)
        dist[e[i].v]=min(dist[e[i].v],e[i].cost);
    while(num<n){
        int res=INF,pos;
        for(int i=1;i<=n;i++)
            if(dist[i]<res&&!vis[i])
                pos=i,res=dist[i];
        vis[pos]=true;
        sum+=res;
        for(int i=first[pos]; i!=-1;i=e[i].next){
            int v=e[i].v;
            if(vis[v]) continue;
            dist[v] = min(dist[v], e[i].cost);
        }
        num++;
    }
    return sum;
}
int main(){
    while(scanf("%d",&n)==1&&n){
        memset(first,-1,sizeof(first));
        ecnt=-1;
        memset(e,0,sizeof(e));
        int a,b,c;
        for(int i=1;i<=(n*(n-1))/2;i++){
            scanf("%d%d%d",&a,&b,&c);
            addedge(a,b,c);
            addedge(b,a,c);
        }
        printf("%d\n",prim());
    }
    return 0;
}

方法三：kruscal

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct edge{
    int v,u,cost;
}e[111*111];
bool cmp(const edge &a,const edge &b){
    return a.cost<b.cost;
}
int ecnt,fa[111],n,v;
void addedge(int u,int v,int cost){
    e[++ecnt].v = v;
    e[ecnt].u = u;
    e[ecnt].cost = cost;
}
int getf(int x){
    return fa[x]==x?x:fa[x]=getf(fa[x]);
}
int kruscal(){
    int sum=0;
    for(int i=1;i<=n;i++)
        fa[i]=i;
    sort(e+1,e+1+ecnt,cmp);
    for(int i=1;i<=ecnt;i++){
        int u=e[i].u,v=e[i].v;
        u=getf(u);
        v=getf(v);
        if(u==v) continue;
        sum+=e[i].cost;
        fa[u]=v;
    }
    return sum;
}
int main(){
    while(scanf("%d",&n)==1&&n){
        ecnt=0;
        int a,b,c;
        for(int i=1;i<=(n*(n-1))/2;i++){
            scanf("%d%d%d",&a,&b,&c);
            addedge(a,b,c);
            addedge(b,a,c);
        }
        printf("%d\n",kruscal());
    }
    return 0;
}