历届试题 城市建设 （两次Kruskal）

问题描述

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

题解：最小生成树啊。

用把所有码头都连到0点。

然后跑两次Kruskal。

第一次跑有码头的图，统计此时最小生成树的边权ans1。

第二次跑无码头的图，统计此时最小生成树的边权ans2。

跑完第二次后，如果刚好还有一个点没有改变fa[i]=i，即还有一个点没有没有加入集合中。(想想kruskal判有环的时候)，此时就去min(ans1,ans2)。否则就去有码头时就是最小花费。

注意：小于0的边权一定要加上，因为只赚不亏啊！

代码：

#include<bits/stdc++.h>
using namespace std;
int read()  
{  
    int v = 0, f = 1;  
    char c =getchar();  
    while( c < 48 || 57 < c ){  
        if(c=='-') f = -1;  
        c = getchar();  
    }  
    while(48 <= c && c <= 57)   
        v = v*10+c-48, c = getchar();  
    return v*f;  
}  
int n,m;

struct edge
{
	int u,v,val;
}e[123456];

int fa[100100];
int ans1,ans2;
int cmp(edge a, edge b)  
{  
    return a.val<b.val;  
}  
int find(int x)
{
	return fa[x]==x? x:fa[x]=find(fa[x]);
}

void kruskal_first() //有码头
{
	//int cnt=n;
	for(int i=1;i<=n;i++)fa[i]=i;
	sort(e,e+n+m,cmp);
	ans1=0;
	ans2=0;
	for(int i=0;i<n+m;i++)
	{
		if(e[i].val == - 0x3f3f3f3f)continue;
		int t1=find(e[i].u);
		int t2=find(e[i].v);
		
		if(t1!=t2|| e[i].val < 0) //少于0的一定要连上，因为是赚钱了！
		{
			
			ans1 += e[i].val;
			fa[t1]=t2; 
			//cnt--;
			//if(cnt==1)break;
		}
	}	
}
void kruskal_second() //无码头
{
	//int cnt=n;
	for(int i=1;i<=n;i++)fa[i]=i;
	sort(e,e+n+m,cmp);
	for(int i=0;i<n+m;i++)
	{
		if(e[i].u == 0) continue;
		int t1=find(e[i].u);
		int t2=find(e[i].v);
		if(t1!=t2 || e[i].val < 0) //少于0的一定要连上，因为是赚钱了！
		{
			ans2 += e[i].val;
			fa[t1]=t2;
		    //cnt--;
			//if(cnt==1)break;
		}
		
	}
}

int main()
{
	ios::sync_with_stdio(false);
    //freopen("in.txt","r",stdin);
	cin>>n>>m;
	for(int i=0;i<m;i++){
		cin>>e[i].u>>e[i].v>>e[i].val;
		
	}
	for(int i=m;i<n+m;i++)//把所有码头都连到一个0点
	{
		e[i].u = 0;
		e[i].v = i - m + 1;
		cin>>e[i].val;	
		if(e[i].val == -1) e[i].val = -0x3f3f3f3f;
	}

	kruskal_first();
	kruskal_second();
	//cout<<min(ans1,ans2)<<endl;
	 
	int sum=0;
	for(int i=1;i<=n;i++){ //跑完无码头后 
		if(fa[i]==i)sum++;
	}
	//cout<<ans1<<" "<<ans2<<endl;
	if(sum==1)cout<<min(ans1,ans2)<<endl;
	else cout<<ans1<<endl;
	
	return 0;
}