poj2421(kruskal)

最新推荐文章于 2020-07-15 12:06:03 发布

张亚军

最新推荐文章于 2020-07-15 12:06:03 发布

阅读量624

点赞数

文章标签： poj

本文链接：https://blog.csdn.net/u014236804/article/details/38081459

版权

题目大意

输入村庄数N<=100，接下来是N*N矩阵（i，j）表示i到j距离

接下来是Q(<=N*(N+1)/2),Q行a，b表示a，b有路；

输出公路长度的最小值

kruskal算法

第一次数组开小了

第二次处理已有边方式错了

#include <iostream>
#include<cstdio>
#include<algorithm>

#define   MAXN  110
#define   MAXM  10000

using namespace std;

int   r[MAXM],u[MAXM],v[MAXM],w[MAXM],p[MAXN];

int cmp(const int i,const int j)
{
    return w[i]<w[j];
}

int Find(int x)
{
    return p[x]==x?x:p[x]=Find(p[x]);
}

int main()
{
    int n,t=0;
    scanf("%d",&n);
    for(int i=0;i<=n;i++)  p[i]=i;
    for(int i=1;i<=n;i++)
    {
        for(int j=1;j<=n;j++)
        {
            int ww;
            scanf("%d",&ww);
            if(i!=j)
            {
            u[t]=i;v[t]=j;w[t]=ww;
            t++;
            }
        }
    }
    for(int i=0;i<t;i++)  r[i]=i;
    sort(r,r+t,cmp);
    int m;
    scanf("%d",&m);
    while(m--)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        int A=Find(a);
        int B=Find(b);
        p[B]=A;
    }
    int ans=0;
    for(int i=0;i<t;i++)
    {
        int e=r[i];int x=Find(u[e]);int y=Find(v[e]);
        if(x!=y)
        {
            ans+=w[e];p[x]=y;
        }
    }
    printf("%d\n",ans);
    return 0;
}