码迷,mamicode.com
首页 > 其他好文 > 详细

还是畅通工程——最小生成树模板题

时间:2019-10-02 12:48:39      阅读:88      评论:0      收藏:0      [点我收藏+]

标签:lan   sort   str   close   stream   pid   +=   blank   one   

题目链接

题意:

给定n个村庄,m=(n*(n-1)/2)条关系  u,v,w 表示  u到 v之间的距离是 w

题解:

裸最小生成树模板题

 

代码:

技术图片
#include<iostream>
#include<stdio.h>
#include<math.h>
#include<algorithm>
#include<vector>
using namespace std;
typedef long long ll;
const int maxn = 10005;
int f[maxn];
int x[maxn],y[maxn];
int n,cnt,m;
struct node
{
    int u,v;
    int w;
    bool operator < (const node &a)const
    {
        return w<a.w;
    }
} edge[maxn*maxn];

int Find(int x)
{
    return x==f[x]?x:f[x]=Find(f[x]);
}
void add(int u,int v,int w)
{
    edge[cnt].u=u;
    edge[cnt].v=v;
    edge[cnt++].w=w;
}

int kruskal()
{
    int ans=0;
    for(int i=0; i<=n; i++)f[i]=i;
    sort(edge,edge+cnt);
    for(int i=0; i<cnt; i++)
    {
        int x=edge[i].u;
        int y=edge[i].v;
        int fx=Find(x);
        int fy=Find(y);
        if(fx!=fy)
        {
            f[fx]=fy;
            ans+=edge[i].w;
        }
    }
    return ans;
}

int main()
{
    while(scanf("%d",&n) && n)
    {
        cnt=0;
        m=n*(n-1)/2;
        for(int i=1; i<=m; i++)
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
        }
        printf("%d\n",kruskal());
    }



    return 0;
}
kruskal

 

 

还是畅通工程——最小生成树模板题

标签:lan   sort   str   close   stream   pid   +=   blank   one   

原文地址:https://www.cnblogs.com/j666/p/11617015.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!