城市建设 最小生成树

问题 1437: [蓝桥杯][历届试题]城市建设

时间限制: 1Sec 内存限制: 128MB 提交: 408 解决: 82

题目描述

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

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

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

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

样例说明
建设第2、3、4条道路，在地点4、5建设码头，总的花费为9。
数据规模和约定
对于100%的数据，1  < =  n  < =  10000，1  < =  m  < =  100000，-1000< =c< =1000，-1< =w_i< =1000，w_i≠0。
 

输入

输入的第一行包含两个整数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

思路：

要求城市联通，可以想到最小生成树。关键是码头怎么用。先不用码头跑一遍最小生成树，再用码头跑一遍生成树。

这里加一个超级源点0，使得0连接所有可能的码头。若不用码头不连通，那么就必须用码头。若联通，就取两者最小值。

代码：

#include<iostream>
#include<algorithm>
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define ll long long
#include<string.h>
#include<queue>
#include<stack>
#include<cstdio>
using namespace std;
const int maxn=300000+6;
#define mod 1000000007
#define INF 0x3f3f3f3f
struct node
{
    int u,v,w;
} a[maxn];
bool cmp(node a,node b)
{
    return a.w<b.w;
}
int wi[maxn];
int vis[maxn];
struct Kruskal
{
    int tot;
    int n;
    int f[maxn];
    int ma;
    void init(int n1)
    {
        //tot=0;
        ma=0;
        n=n1;
        rep(i,0,n)f[i]=i;
    }
    void add(int u,int v,int w)
    {
        ++tot;
        a[tot].u=u;
        a[tot].v=v;
        a[tot].w=w;
    }
    int finds(int x)
    {
        return x==f[x]?x:f[x]=finds(f[x]);
    }
    int work()
    {
        sort(a+1,a+tot+1,cmp);
        int cnt=0;
        int sum=0;
        for(int i=1; i<=tot; i++)
        {
            int u=a[i].u;
            int v=a[i].v;
            int w=a[i].w;
            int fu=finds(u);
            int fv=finds(v);
            if(fu!=fv)
            {
                f[fu]=fv;
                cnt++;
                sum+=w;
            }else if(w<0)
            {
                sum+=w;
            }
        }
        return sum;
    }
    bool ok()
    {
        int cnt=0;
        int root=finds(1);
        for(int i=2; i<=n; i++)
        {
            if(finds(i)!=root)
                cnt++;
        }
        return cnt==0?1:0;
    }
} Kru;
int main()
{
    int n,m;
    cin>>n>>m;
    Kru.init(n);
    Kru.tot=0;
    rep(i,1,m)
    {
        int u,v,w;
        cin>>u>>v>>w;
        Kru.add(u,v,w);

    }
    rep(i,1,n)
    {
        cin>>wi[i];
    }
    int ans=Kru.work();
    int flag=0;
    if(!Kru.ok())
        flag=1;//不连通
    rep(i,1,n)
    {
        if(wi[i]>=0)
        {
            Kru.add(0,i,wi[i]);
        }
    }
    Kru.init(n+1);
    if(flag)//不通
        cout<<Kru.work()<<endl;
    else//通了
    {
        cout<<min(ans,Kru.work())<<endl;
    }
    return 0;
}