【BUAA 595】太空漫步

来源：互联网发布：汽车怎么设计软件编辑：程序博客网时间：2024/04/27 17:47

【BUAA 595】太空漫步

最小生成树

kruskal模板题

代码如下

#include <bits/stdc++.h>using namespace std;typedef struct Edge{    int u,v,d;    bool operator < (const struct Edge a)const    {        return d < a.d;    }}Edge;Edge eg[111111];int pre[111];int n,tp;void Init(){    int i;    for(i = 1; i <= n; ++i)        pre[i] = i;}int Find(int x){    if(x != pre[x]) pre[x] = Find(pre[x]);    return pre[x];}void Add(int u,int v,int d){    eg[tp].u = u;    eg[tp].v = v;    eg[tp++].d = d;}int main(){    scanf("%d",&n);    tp = 0;    int i,j,d,cnt,sum,k,r;    Init();    for(i = 1; i <= n; ++i)    {        for(j = 1; j <= n; ++j)        {            scanf("%d",&d);            if(i < j)            {                Add(i,j,d);            }        }    }        sort(eg,eg+tp);        cnt = sum = 0;        for(i = 0; i < tp; ++i)        {            k = Find(eg[i].u);            r = Find(eg[i].v);            if(k != r)            {                pre[k] = r;                sum += eg[i].d;                cnt++;            }            if(cnt == n-1) break;        }        printf("%d\n",sum);    return 0;}

0 0