poj2421

题目大意:有n个村庄,要求把n个村庄连接在一起的最短距离

解题思路:kruskal算法即可,注意就是有点村庄已经连载一起了,把邻接矩阵的距离变为0就行了

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
using namespace std;
const int maxs=100+10;
const int maxn=100*50+100;
int G[maxs][maxs];
int u[maxn],v[maxn],w[maxn],p[maxn],r[maxn],n,m;//数组u储存起始点,数组v存储终点,数组w存储边权,m是边数,n是点数
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 kruskal()
{
    int cou=0,x,y,i,ans=0;
    for(i=1;i<=n;i++) p[i]=i;//初始化并查集
    for(i=1;i<=m;i++) r[i]=i;
    sort(r+1,r+m+1,cmp);//根据边权排序
    for(i=1;i<=m;i++)
    {
        int e=r[i];x=find(u[e]);y=find(v[e]);
        if(x!=y) {ans += w[e];p[x]=y;cou++;}
        if(cou==n-1)  {return ans;}
    }
    if(cou<n-1) ans=0;//n个点的数有n条边
    return ans;
}
int main()
{
        while(scanf("%d",&n)!=EOF)
        {
            for(int i=1;i<=n;i++)
                for(int j=1;j<=n;j++)
                scanf("%d",&G[i][j]);
               int q;
               scanf("%d",&q);
               for(int i=1;i<=q;i++)
               {
                       int x,y;
                       scanf("%d%d",&x,&y);
                       G[x][y]=0;
                       G[y][x]=0;
               }
               m=0;
               for(int i=1;i<=n;i++)
                     for(int j=i+1;j<=n;j++)
                     {
                             m++;
                             u[m]=i;
                             v[m]=j;
                             w[m]=G[i][j];

                     }
            printf("%d\n",kruskal());
        }
        return 0;
}



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值