1732: 数花费（Kruscal）

1732: 数花费
时间限制: 1 Sec 内存限制: 128 MB

[提交][状态][讨论版]
题目描述
　栋栋居住在一个繁华的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。
第一遍跑没有码头的。
第二遍跑加上码头的。
对于第一遍没有码头的而言，可能无法生成最小生成树。
判断条件就是: 根节点（pre[i] == i）个数是否等于1
不管带不带码头跑生成树，有钱赚的肯定建它啊！！
对于第二遍，由于城市数量最大达10000，码头之间的边无法暴力处理。
此时可以引入一个虚拟节点0，让所有可以建码头的城市通过节点0连接起来。
最后算答案的时候，如果没有码头无法连通，那么就直接取有码头的那个答案，
否则取两者之间较小值。
*/
Ac_code:

#include <bits/stdc++.h>

using namespace std;
const int maxn= 1e5+5;
#define INF 0x3f3f3f3f
struct Edge
{
    int s,e;
    int val;
    bool operator<(const Edge a)const
    {
        return val < a.val;
    }
} data[maxn<<1];
int pre[maxn];
int myFind(int x)
{
    int r = x;
    while(pre[r]!= r)
    {
        r = pre[r];
    }
    int i = x,j;
    while(pre[i]!= r)
    {
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;
    //return pre[x]==x?x:pre[x]=myFind(pre[x]);
}
int kruscal(int n,int m)
{
    for(int i = 1; i <= n; i++)
    {
        pre[i] = i;
    }
    sort(data,data+m);
    int ans = 0;
    for(int i = 0; i < m; i++)
    {
        if(data[i].val == INF) continue; //不能建码头
        int fx = myFind(data[i].s);
        int fy = myFind(data[i].e);
        if(fx != fy || data[i].val < 0)//赚钱的路一定建
        {
            ans += data[i].val;
            pre[fx] = fy;
        }
    }
    return ans;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i = 0; i < m; i++)
    {
        scanf("%d%d%d",&data[i].s,&data[i].e,&data[i].val);
    }
    int t;
    for(int i = m; i < n+m; i++)
    {
        scanf("%d",&t);
        data[i].s = 0; //通过虚拟一个0点连接码头之间构成的边
        data[i].e = i-m+1;
        if(t == -1)
            data[i].val = INF;
        else
            data[i].val = t;
    }
    int ans1 = kruscal(n,m); //无码头
    int cnt = 0;
    for(int i = 1; i <= n; i++)//判断无码头能否连通
    {
        if(pre[i]==i) cnt++;
    }
    int ans2 = kruscal(n,m+n);//有码头
    printf("%d\n",(cnt==1?min(ans1,ans2):ans2));
    return 0;
}