hdu 1879 继续畅通工程 最小生成树kruskal

题目链接

题意：给出n个点，(n-1)*n/2条路，其中有若干条已经建好。求最小生成树。

这题用prim应该效果比较好，但kruskal已经很生疏了，这次给出kruskal模板。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define N 110
#define M 6600

using namespace std;

struct node
{
    int u,v,w;
}e[M];

int pre[N],n,cnt;

bool cmp(node a,node b)
{
    return a.w<b.w;
}

void add_edge(int u,int v,int w)
{
    e[cnt].u=u;
    e[cnt].v=v;
    e[cnt++].w=w;
}

int findset(int v)
{
    while(v!=pre[v])    v=pre[v];
    return v;
}

void kruskal()
{
    sort(e,e+cnt,cmp);
    for(int i=1;i<=n;i++)   pre[i]=i;
    int t=0,ans=0;
    for(int i=0;i<cnt;i++)
    {
        int u=e[i].u,v=e[i].v,w=e[i].w;
        int t1=findset(u);
        int t2=findset(v);
        if(t1!=t2)
        {
            ans+=w;
            pre[t1]=t2;
            t++;
        }
        if(t==n-1)  break;
    }
    cout<<ans<<endl;
}

int main()
{
    while(~scanf("%d",&n)&&n)
    {
        cnt=0;
        int m=(n-1)*n/2;
        for(int i=0;i<m;i++)
        {
            int u,v,w,t;
            scanf("%d%d%d%d",&u,&v,&w,&t);
            if(t)   w=0;
            add_edge(u,v,w);
        }
        kruskal();
    }
}