hdu 1102 Constructing Roads 最小生成树Kruskal

这原本是最小生成树prim部分的题,但感觉kruskal好用,就用kruskal 打了

这道题题意有点变扭,修路,使所有村庄连通,使修路的街道总长度最小;原本还以为使街道的总长度最小,但案例过不了才发觉

使已经修了的路的权值为0,其他路的权值当然为路的长度,模板题

代码:

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int n;
const int maxn=10010;
int a[maxn][maxn];
int fa[maxn];
struct Node
{
    int u;
    int v;
    int weight;
};
Node b[maxn];
bool cmp(Node x,Node y)
{
    return x.weight<y.weight;
}
void init()
{
    int i;
    for(i=1;i<=n;i++)
    {
        fa[i]=i;
    }
}
int find(int u)
{
    return u==fa[u]?fa[u]:fa[u]=find(fa[u]);
}
void join(int u,int v)
{
    fa[find(u)]=find(v);
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        int i,j;
        init();
        int start=0;
        for(i=1;i<=n;i++)
        {
            for(j=1;j<=n;j++)
                scanf("%d",&a[i][j]);
        }
        int q;
        scanf("%d",&q);
        int u,v;
         while(q--)
        {
            scanf("%d%d",&u,&v);
            a[u][v]=0;
            a[v][u]=0;
        }
         for(i=1;i<=n;i++)
        {
            for(j=1;j<i;j++)
            {
                b[start].u=i;
                b[start].v=j;
                b[start++].weight=a[i][j];
            }
        }
        sort(b,b+start,cmp);
        int cnt=0;
        int ans=0;
        for(i=0;i<start;i++)
        {
           int u=b[i].u;
           int v=b[i].v;
           if(find(u)!=find(v))
           {
               ans+=b[i].weight;
               join(u,v);
               cnt++;
           }
           if(cnt==n-1)
            break;
        }
        printf("%d\n",ans);
    }
    return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值