城市建设 蓝桥真题

 

最小生成树变形 变在怎么处理码头 显然不能两码头的建设费用作为一条边 可以建立一个虚点 将点的码头建设费用作为边权与其相连

但是虚点也是点 如果需要走水路那图中就有 (n+1)个点 不需要的话还是n个点 此处要分类讨论

1. 所有城市可以通过陆路相连

此时可能只走陆路花费少 走水路反而费钱 先对这n个点求最小生成树

也可能需要水路松弛 那就再求(n+1)个点的最小生成树

两者取最小值

2. 所有城市必须通过一些水路相连

此时n个点的生成树(即只走陆路 不过虚点)不存在 最优解只会在(n+1)个点的生成树里

 

 

#include <bits/stdc++.h>
using namespace std;
#define N 0x3f3f3f3f

struct node
{
    int u;
    int v;
    int w;
};

node edge[110010];
int f[10010],dock[10010];
int n,m,mm;

int getf(int p)
{
    if(f[p]==p) return p;
    else
    {
        f[p]=getf(f[p]);
        return f[p];
    }
}

bool unite(int u,int v)
{
    int fu,fv;
    fu=getf(u);
    fv=getf(v);
    if(fu!=fv)
    {
        f[fv]=fu;
        return true;
    }
    return false;
}

bool cmp(node n1,node n2)
{
    return n1.w<n2.w;
}

int main()
{
    int i,sum1,sum2,cnt,mis,tem;
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=1;i<=n;i++)
        {
            f[i]=i;
        }
        mis=0;
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
            unite(edge[i].u,edge[i].v);
            if(edge[i].w<0)
            {
                mis+=edge[i].w;
            }
        }
        mm=m;
        for(i=1;i<=n;i++)
        {
            scanf("%d",&dock[i]);
            if(dock[i]!=-1)
            {
                mm++;
                edge[mm].u=0,edge[mm].v=i,edge[mm].w=dock[i];
            }
        }
        cnt=0;
        for(i=1;i<=n;i++)
        {
            if(f[i]==i) cnt++;
        }
        if(cnt==1)
        {
            sort(edge+1,edge+m+1,cmp);
            for(i=1;i<=n;i++)
            {
                f[i]=i;
            }
            sum1=0,cnt=0,tem=mis;
            for(i=1;i<=m;i++)
            {
                if(unite(edge[i].u,edge[i].v))
                {
                    sum1+=edge[i].w,cnt++;
                    if(edge[i].w<0) tem-=edge[i].w;
                }
                if(cnt==n-1) break;
            }
            sum1+=tem;
        }
        else sum1=N;
        sort(edge+1,edge+mm+1,cmp);
        for(i=0;i<=n;i++)
        {
            f[i]=i;
        }
        sum2=0,cnt=0,tem=mis;
        for(i=1;i<=mm;i++)
        {
            if(unite(edge[i].u,edge[i].v))
            {
                sum2+=edge[i].w,cnt++;
                if(edge[i].w<0) tem-=edge[i].w;
            }
            if(cnt==n) break;
        }
        sum2+=tem;
        printf("%d\n",min(sum1,sum2));
    }
    return 0;
}