【Kruscal】城市建设

问题描述

　　栋栋居住在一个繁华的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，向可以建码头的点连边，边权为建码头的费用。
在只建道路所有点就能互达的情况下：
  为了使0点联通可能会使费用增加，所以要在建码头与不建码头两种情况中取较小值。
在只建道路所有点不能互达的情况下：
   答案就为建码头的最小费用

代码:

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn = 10000 + 10;
const int maxm = 100000 + 10;

struct Edge
{
	int s,t,w;
}edge[maxm << 1];
int we[maxn], f[maxn];
int n,m;
int tmp = 0;

void init()
{

}

int cmp(const Edge &a,const Edge &b)
{
	return a.w < b.w;
}

int max(int a,int b)
{
	return a > b ? a : b;
}

void readdata()
{
	scanf("%d%d", &n, &m);
	for(int i = 1;i <= m;i++)
	{
		scanf("%d%d%d", &edge[i].s, &edge[i].t, &edge[i].w);
	}
	for(int i = 1;i <= n;i++)scanf("%d", &we[i]);
	for(int i = 1;i <= n;i++)
	{
		if(we[i] == -1)continue;
		tmp++;
		edge[++m].s = 0;
		edge[m].t = i;
		edge[m].w = we[i];
	}
}

int find(int x)
{
	if(f[x] == x)return x;
	f[x] = find(f[x]);
	return f[x];
}

void merge(int x, int y)
{
	int wx = find(x);
	int wy = find(y);
	f[wx] = wy;
}

void predo()
{
	for(int i = 0;i <= n;i++)f[i] = i;
}

bool test()
{
	predo();
	for(int i = 1;i <= m;i++)
	{
		int s = edge[i].s, t = edge[i].t;
		if(s == 0)continue;
		if(find(s) != find(t))merge(s, t);
	}
	bool ret = true;
	for(int i = 2;i <= n;i++)
	{
		if(find(i) != find(i-1))
		{
			ret = false;
			break;
		}
	}
	return ret;
}

int Kruscal(bool flag)
{
	predo();
	sort(edge + 1, edge + m + 1, cmp);
	int ret = 0;
	for(int i = 1;i <= m;i++)
	{
		int ps = edge[i].s, pt = edge[i].t, pw = edge[i].w;
		if(!flag && ps == 0)continue;
		if(find(ps) != find(pt) || pw < 0)
		{
			merge(ps, pt);
			ret += pw;
		}
	}
	return ret;
}

void solve()
{
	bool link = test();
	int ans;
	if(link)
	{
		ans = min(Kruscal(false), Kruscal(true));
	}
	else ans = Kruscal(true);
	printf("%d\n", ans);
	
}

int main()
{
	init();
	readdata();
	solve();
	return 0;
}