蓝桥杯 历届试题 城市建设 最小生成树

把码头作为0点处理。

首先判断不建码头是否可以生成最小生成树

最小生成树用kruskal算法，若对于代价<0的边，直接加入

若可以：Min（最小生成树（不建码头），最小生成树（建码头））；

若不可：最小生成树（建码头）

#include "stdio.h"
#include "string.h"
#include "algorithm"
using namespace std;
int father[10010];
struct Link
{
    int a,b,v;
}link[110001];

int Min(int a,int b)
{
    if (a<b) return a;
    else return b;
}
int fin(int x)
{
    if (x==father[x]) return x;
    father[x]=fin(father[x]);

    return father[x];
}

bool cmp(Link a,Link b)
{
    return a.v<b.v;
}

int kruskal(int m)
{
    int sum,i,fa,fb;
    sum=0;
    for (i=0;i<m;i++)
    {
        fa=fin(link[i].a);
        fb=fin(link[i].b);
        if (fa!=fb || link[i].v<0)
        {
            sum+=link[i].v;
            father[fa]=fb;
        }
    }
    return sum;
}


int main()
{
    int n,m,i,cnt,fa,fb,ans,x;
    scanf("%d%d",&n,&m);
    for (i=0;i<=n;i++)
        father[i]=i;

    for (i=0;i<m;i++)
        scanf("%d%d%d",&link[i].a,&link[i].b,&link[i].v);

    cnt=0;
    for (i=1;i<=n;i++)
    {
        scanf("%d",&x);
        if (x!=-1)
        {
            link[m+cnt].a=0;
            link[m+cnt].b=i;
            link[m+cnt].v=x;
            cnt++;
        }
    }

    for (i=0;i<m;i++)
    {
        fa=fin(link[i].a);
        fb=fin(link[i].b);
        father[fa]=fb;
    }

    for (i=2;i<=n;i++)
        if (fin(1)!=fin(i))
        break;

    if (i==n+1)
    {
        for (i=0;i<=n;i++)
            father[i]=i;
        sort(link,link+m,cmp);
        ans=kruskal(m);

        for (i=0;i<=n;i++)
            father[i]=i;
        sort(link,link+m+cnt,cmp);
        ans=Min(ans,kruskal(m+cnt));
        printf("%d\n",ans);
    }
    else
    {
        for (i=0;i<=n;i++)
            father[i]=i;
        sort(link,link+m+cnt,cmp);
        printf("%d\n",kruskal(m+cnt));
    }
    return 0;


}