hdu 1233 还是畅通工程(最小生成树)

还是畅通工程

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

Problem Description

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

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

Output

对每个测试用例，在1行里输出最小的公路总长度。

Sample Input

Sample Output

3
5


Hint
Hint
 
Huge input, scanf is recommended.


#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define N 11001
using namespace std;

int par[N],Rank[N];
int n,E;
struct edge {
    int u,v,cost;
};

edge es[N];

void init() {
    for(int i=0; i<=n; i++) {
        par[i]=i;
        Rank[i]=0;
    }
}

bool cmp(edge e1,edge e2) {
    return e1.cost<e2.cost;
}

int Find(int x) {
    if(par[x]==x)return x;
    return par[x]=Find(par[x]);
}

void unite(int x,int y) {
    x=Find(x);
    y=Find(y);
    if(x==y)return;
    if(Rank[x]<Rank[y])par[x]=y;
    else {
        par[y]=x;
        if(Rank[x]==Rank[y])Rank[x]++;
    }
}

bool same(int x,int y) {
    return Find(x)==Find(y);
}

int Kruskal() {
    int ans=0;
    for(int i=0; i<E; i++) {
        edge e=es[i];
        if(!same(e.u,e.v)) {
            ans+=e.cost;
            unite(e.u,e.v);
        }
    }
    return ans;
}

int main() {
    while(cin>>n&&n) {
        E=n*(n-1)/2;
        for(int i=0; i<E; i++) {
            scanf("%d%d%d",&es[i].u,&es[i].v,&es[i].cost);
        }
        sort(es,es+E,cmp);
        init();
        int ans=Kruskal();
        cout<<ans<<endl;
    }
    return 0;
}

hdu 1233 仍是畅通工程(最小生成树)

还是畅通工程

相关推荐