【BUAA 933】拮据的模拟城市

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

【BUAA 933】拮据的模拟城市

最小生成树 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[1111];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(){    int i,u,v,d,cnt,sum,k,m,r;    while(~scanf("%d %d",&n,&m))    {        tp = 0;        Init();        while(m--)        {            scanf("%d %d %d",&u,&v,&d);            Add(u,v,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