Airport(最小生成树)

题目:https://vjudge.net/contest/241417#problem/C

分析:如果修路的费用大于建一个机场的费用,那就建一个机场,cnt++,最终的机场数目等于cnt+集合数目。

#include <iostream>
#include <math.h>
#include <stdio.h>
#include <string.h>
#include <algorithm>
#define maxn 10005
#define maxm 100005
#define INF 0x3f3f3f3f
using namespace std;
struct Edge
{
    int u,v;
    int w;
}edge[maxm];//存储边信息。
int tol;//边数.加边前赋值0;
int n,m,a;
int pre[maxn],ran[maxn],ans,cnt;
void addedge(int u,int v,int w)
{
    edge[tol].u=u;
    edge[tol].v=v;
    edge[tol++].w=w;
}
//排序函数,将边按照权值由小到大排序。
bool cmp(Edge a,Edge b)
{
    return a.w<b.w;
}
int  find(int x)
{
    return pre[x]==x?x:pre[x]=find(pre[x]);
}
void Union(int x,int y)
{
    if(ran[x]<ran[y])pre[x]=y;
    else if(ran[x]==ran[y]){ran[x]++;pre[y]=x;}
}
//传入点数,返回最小生成树的权值,如果不联通返回-1;
void kruskal()
{
    sort(edge,edge+tol,cmp);
    cnt=0;//计算加入边的数目。
    ans=0;
    for(int i=0;i<tol;i++)
    {
        int u=edge[i].u;
        int v=edge[i].v;
        int w=edge[i].w;
        int t1=find(u);
        int t2=find(v);
        if(t1!=t2)
        {
            Union(t1,t2);
            ans+=w;
        if(w>=a){
                ans-=w;
             cnt++;
          }
        }
    }
}
void solve()
{
    for(int i=1;i<=n;i++)
        if(find(i)==i)cnt++;
        ans+=a*cnt;
}
int main()
{
    int t;
    scanf("%d",&t);
    int cas=0;
    while(t--)
    {
        int x,y,cost;
        scanf("%d%d%d",&n,&m,&a);
        for(int i=1;i<=n;i++)
        {
            pre[i]=i;ran[i]=1;
        }
        tol=0;
       for(int i=1;i<=m;i++)
       {
           scanf("%d%d%d",&x,&y,&cost);
           addedge(x,y,cost);
       }
       kruskal();
       solve();
       printf("Case #%d: %d %d\n",++cas,ans,cnt);
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/guagua_de_xiaohai/article/details/81229980
今日推荐