K - 还是畅通工程 (HDU1233)-kruskal

某省调查乡村交通状况,得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通(但不一定有直接的公路相连,只要能间接通过公路可达即可),并要求铺设的公路总长度为最小。请计算最小的公路总长度。 
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
这题是kruskal算法,模板题吧 ,可以用快排 也能用sort。


#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
struct edge
{
    int u,v,w;
}map[10000];
int father[10000]= {0},sum=0,ans=0;

void quicksort(int left,int right)
{
    int i,j;
    struct edge t;
    if(left>right)
        return;

    i=left;
    j=right;
    while(i!=j)
    {
        while(map[j].w>=map[left].w&&i<j)
            j--;
        while(map[i].w<=map[left].w&&i<j)
            i++;
        if(i<j)
        {
            t=map[i];
            map[i]=map[j];
            map[j]=t;
        }
    }
    t=map[left];
    map[left]=map[i];
    map[i]=t;
    quicksort(left,i-1);
    quicksort(i+1,right);
    return ;
}
int cmp(edge a,edge b)
{
    return a.w<b.w;
}
int find_father(int x)
{
    if(x==father[x]) return x;
    else
    {
        father[x]=find_father(father[x]);
        return father[x];
    }
}
int merge(int u,int v)
{
    int x=find_father(u);
    int y=find_father(v);
    if(x!=y)
    {
        father[y]=x;
        return 1;
    }
    return 0;
}

int main()
{
    int n,m;
    while(~scanf("%d",&n))//顶点
    {
        if(n==0) break;
        sum=0;
        m=n*(n-1)/2;//边
        for(int i=1; i<=m; i++)
        {
            scanf("%d%d%d",&map[i].u,&map[i].v,&map[i].w);
        }
        quicksort(1,m);
        //sort(map+1,map+m+1,cmp);
        for(int i=1; i<=n; i++)
        {
            father[i]=i;
        }
        for(int i=1; i<=m; i++)
        {
            if(merge(map[i].u,map[i].v))
            {
                ans++;
                sum+=map[i].w;
            }
        }
        printf("%d\n",sum);

    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章