蓝桥杯 城市建设（kruscal变形）

历届试题 城市建设  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

　　栋栋居住在一个繁华的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。

tips：

1.如果有某俩个结点不通，我们可以通过在这俩个地方架设码头，只要有码头的地方都是相通的，标注为-1的地方不能修建码头。

2：怎么处理架设码头的点呢？我们可以另外架设一个结点0，将所有假设码头的点和0结点相连然后在处理；

3：因为架设码头会多出一个点，所以我们要分开判断，以防止结点数的判断上出问题，这是因为：

1）.如果不假设码头不联通的话，那么意味着必须要经过码头（即经过结点0），那么直接算出n+1个点的最小生成树即可。

2）.如果不架设码头也联通的话，此时对应两种情况：一。不走码头费用最少，二。经过码头费用最小，此时应取min(n,n+1)

那么为什么合并在一起算结果可能不对呢，这时就对应了结果上面中的情况2，此时的最小值应该是不走码头，但是要求n+1各点的最小生成树，又一定会经过码头，结果变大，这就是造成错误结果的原因。

4：最坑的是有的路径可以盈利，为了使花费最小，这些路径不管怎么样都要加上，稍微变形一下；

#include<iostream>#include<cstring>#include<vector>#include<algorithm>using namespace std;struct edge{int u,v,w;friend bool operator <(edge e1,edge e2){return e1.w<e2.w;}};vector<edge>edges;int n,m,ans;int cost[11000];int f[11000];int find(int x){return f[x]<0?x:f[x]=find(f[x]);}bool merge(int x,int y){int rx=find(x);int ry=find(y);if(rx!=ry){f[rx]+=f[ry];f[ry]=rx;return true; }return false;}int  kruscal(int nn,int mm)//传入的点数和边数 {sort(edges.begin(),edges.begin()+mm);memset(f,-1,sizeof(f));int cnt,sum;cnt=sum=0;for(int i=0;i<mm;i++){int x=edges[i].u;int y=edges[i].v;if(merge(x,y)){sum+=edges[i].w;++cnt;//if(++cnt==nn-1)break;}//注意和kruscal的这里区别 else{if(edges[i].w<0){sum+=edges[i].w;}}} if(cnt<nn-1)return 0;else return sum;}int main(){cin>>n>>m;for(int i=1;i<=m;i++){int x,y,z;cin>>x>>y>>z;edges.push_back(edge{x,y,z});}int t,cc=0;for(int i=1;i<=n;i++){cin>>cost[i];if(cost[i]!=-1){cc++;edges.push_back(edge{0,i,cost[i]});}}if(t=kruscal(n,m)){ans=min(t,kruscal(n+1,m+cc)); cout<<ans<<endl;}else cout<<kruscal(n+1,m+cc);return 0; }