题目 1437: 蓝桥杯历届试题-城市建设

蓝桥杯的题好难啊。要了命了要

---

连接两个点，可以用道路，也可以用运河。
思路即把运河想象成虚拟节点0。然后按kruscal按边权从小到大顺序排列即可。
不过这个题是求最小生成树的一个变式，求的不是最小生成树，如果权值<0说明赚钱，即无论是否构成环都应该加进去；所以这个地方的循环边界就不是构造n-1条边或者加入0结点构造n条边，而是循环遍历整个存储边的集合，先把负的直接存进去，然后再把正的存进去，正的就需要管是否能构成环。
第二点就是关于运河。他要连接的话，至少要连接两个点，即0结点至少要与两个点相连，因为0本身不是一个点而是运河是承上启下的作用，所以如果连接运河的节点个数为1，则要再减去该节点和0结点的边权值。（对应自己的思路如果i==n&&tag=1,找到了n条边但是0结点没有加进去）这样直接减是正确的是因为0只连接了一个节点，0连接的这个节点肯定还和别的节点相连，所以和0结点相连本身就是多余的，因为此时已经构成最小生成树，如果0连接了>=2个结点，说明这>=2个结点通过运河连接比通过直接建公路便宜，而此时0只连接了一个结点，说明直接建公路更便宜，所以连接0是多余的。

---

时隔接近两个月在做，发现就没那么难了
就是把运河想象成超级源点0，然后后面的n个就把运河和第i个结点连起来。

重点是需要看有没有放0。最后是把所有的点都连接起来，0也会和其他点连接起来。如果运河只连了一次，说明后面连接0的花费是额外的不需要这个花费，要减去这个花费。

一定要自己再做一遍。代码也简洁了思路也清晰了。
用0 和不用0确实需要好好想想才能想出来。

---

又过了4个月再来整理代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;
const int N = 100000 + 10;
int n,m;
int res = 0;    //表示最后的道路的结果。。
int f[N];
vector<int> zero;   //表示和0连接的边的权值的情况。如果在生成最后的路径时只有一个点与0连接，则最后的结果要减去这个点与0之间的权值
//优化的kurscal算法求最小生成树 在找是否会构成环时用到了 并查集优化
void Init()
{
    for(int i=1;i<N;i++) f[i] = i;
}
int find(int x)
{
    if(x == f[x]) return x;
    return f[x] = find(f[x]);
}
void Union(int a,int b)
{
    a = find(a);
    b = find(b);
    f[b] = a;
}

//添加边
int cnt = 0;    //表示的添加的边个条数 总的
struct edge
{
    int u,v,w;
    bool operator<(const struct edge & a)const
    {
        return w < a.w;
    }
}e[N * 2];
void kruscal()
{
    sort(e+1,e+1+cnt);  //一共有 cnt 条边

    for(int i=1;i<=cnt;i++)
    {
        int u,v,w;
        u = e[i].u;
        v = e[i].v;
        w = e[i].w;
        if(find(u) != find(v) || w <= 0)
        {
            Union(u,v);
            res += w;
            if(u==0 || v==0) zero.push_back(w);
        }
    }
    if(zero.size() == 1) res -= zero[0];
    printf("%d",res);
}
int main()
{
    Init();
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        e[++cnt] = {u,v,w};
    }
    //添加运河
    for(int i=1;i<=n;i++)
    {
        int temp;
        scanf("%d",&temp);
        if(temp!=-1)
        {
            e[++cnt] = {0,i,temp};
        }
    }
    kruscal();
}
    int ans = 0;    //最后的花费
    int sum = 0;    //0的个数。

    cin>>n>>m;
    int cnt = 0;
    for(int i=1;i<=m;i++)
    {
        int x,y,w;
        cin>>x>>y>>w;
        e[++cnt].u=x;
        e[cnt].v = y;
        e[cnt].w = w;
    }

    for(int i=1;i<=n;i++)
    {
        int temp;
        cin>>temp;
        if(temp!=-1)
        {
            e[++cnt].u=0;
            e[cnt].v = i;
            e[cnt].w = temp;
        }
    }
    sort(e+1,e+1+cnt);
    for(int i=1;i<=cnt;i++)
    {
        int u,v,w;
        u=e[i].u;
        v=e[i].v;
        w=e[i].w;
        if(find(u)!=find(v) || w < 0)   //只要w < 0 就二话不说直接连好吧
        {
           if(v==0||u==0) zero[++sum] = w;
            ans +=w;
            Union(u,v);
        }
    }
    if(sum==1) ans -=zero[sum];
    cout<<ans;
}