【蓝桥杯】最小生成树 城市建设

问题描述

　　栋栋居住在一个繁华的C市中，然而，这个城市的道路大都年久失修。市长准备重新修一些路以方便市民，于是找到了栋栋，希望栋栋能帮助他。

　　C市中有n个比较重要的地点，市长希望这些地点重点被考虑。现在可以修一些道路来连接其中的一些地点，每条道路可以连接其中的两个地点。另外由于C市有一条河从中穿过，也可以在其中的一些地点建设码头，所有建了码头的地点可以通过河道连接。

　　栋栋拿到了允许建设的道路的信息，包括每条可以建设的道路的花费，以及哪些地点可以建设码头和建设码头的花费。

　　市长希望栋栋给出一个方案，使得任意两个地点能只通过新修的路或者河道互达，同时花费尽量小。

输入格式

　　输入的第一行包含两个整数n, m，分别表示C市中重要地点的个数和可以建设的道路条数。所有地点从1到n依次编号。
　　接下来m行，每行三个整数a, b, c，表示可以建设一条从地点a到地点b的道路，花费为c。若c为正，表示建设是花钱的，如果c为负，则表示建设了道路后还可以赚钱（比如建设收费道路）。
　　接下来一行，包含n个整数w_1, w_2, …, w_n。如果w_i为正数，则表示在地点i建设码头的花费，如果w_i为-1，则表示地点i无法建设码头。
　　输入保证至少存在一个方法使得任意两个地点能只通过新修的路或者河道互达。

输出格式

　　输出一行，包含一个整数，表示使得所有地点通过新修道路或者码头连接的最小花费。如果满足条件的情况下还能赚钱，那么你应该输出一个负数。

样例输入

5 5
1 2 4
1 3 -1
2 3 3
2 4 5
4 5 10
-1 10 10 1 1

样例输出

9

样例说明

　　建设第2、3、4条道路，在地点4、5建设码头，总的花费为9。

数据规模和约定

　　对于20%的数据，1<=n<=10，1<=m<=20，0<=c<=20，w_i<=20；
　　对于50%的数据，1<=n<=100，1<=m<=1000，-50<=c<=50，w_i<=50；
　　对于70%的数据，1<=n<=1000；
　　对于100%的数据，1 <= n <= 10000，1 <= m <= 100000，-1000<=c<=1000，-1<=w_i<=1000，w_i≠0。

参考博客

思路：感觉是个最小生成树但是没想好码头怎么处理，我好菜啊

计算两次Kruscal，一次是没建码头，一次建码头。

关于建码头，引入一个0节点，城市i能建码头就建边 0-i-费用，然后跑Kruscal

注意点1：不能通过边数到达n-1退出Kruscal，因为如果边权为-1，一定要修路（只会使最终费用变小）

注意点2：不建码头的情况下图可能不连通，在不连通的情况下min(sum1,sum2)会出现错误的结果，卡test 8

未A待改

#include<iostream>
#include<algorithm>
using namespace std;
const int M=1e5+10;
const int N=1e4+10;
int fa[N];
int cnt=0;
struct edge
{
    int u,v,w;
    edge(){}
    edge(int u,int v,int w):u(u),v(v),w(w){}
    bool operator <(const edge &rhs)const
    {
        return w<rhs.w;
    }
}e[M];
int n,m;
void init(int n)
{
    for(int i=1;i<=n;++i)
        fa[i]=i;
}
int find(int x)
{
    if(fa[x]==x)
        return x;
    else return fa[x]=find(fa[x]);
}
int kruscal(int m)
{
    init(n);
    int sum=0;
    for(int i=1;i<=m;++i)
    {
        int x=e[i].u,y=e[i].v,w=e[i].w;
        int fx=find(x),fy=find(y);
        if(fx!=fy||w<0)
        {
            sum+=w;
            fa[fx]=fy;
        }
    }
    return sum;                 //忘记return 了，我干啥鸭嘤
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;++i)
        cin>>e[i].u>>e[i].v>>e[i].w;
    cnt=m;
    for(int i=1;i<=n;++i)
    {
        int w;
        cin>>w;
        if(w!=-1)
            e[++cnt]=edge(i,0,w);
    }
   // cout<<cnt<<endl;
    sort(e+1,e+1+m);
    int sum1=kruscal(m);
    int cal=0;
    for(int i=1;i<=n;++i)
        if(fa[i]==i) cal++;
    sort(e+1,e+1+cnt);
    int sum2=kruscal(cnt);
   // cout<<sum1<<' '<<sum2<<endl;
    int ans;
    if(cal==1) ans=min(sum1,sum2);
    else ans=sum2;
    cout<<ans<<endl;
    return 0;


}