蓝桥杯 历届考题 城市建设 （最小生成树）

　　栋栋居住在一个繁华的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无法建设码头。
　　输入保证至少存在一个方法使得任意两个地点能只通过新修的路或者河道互达。

　　输出一行，包含一个整数，表示使得所有地点通过新修道路或者码头连接的最小花费。如果满足条件的情况下还能赚钱，那么你应该输出一个负数。

　　建设第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。

ac代码

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define INF 0xfffffff
int n,m,k;
int pre[10005];
struct s
{
	int u,v,w;
}edge[1000005];
int cmp(const void *a,const void *b)
{
	return (*(struct s *)a).w-(*(struct s *)b).w;
}
int find(int x)
{
	if(x==pre[x])
		return x;
	return pre[x]=find(pre[x]);
}
void init(int n)
{
	int i;
	for(i=0;i<=n;i++)
	{
		pre[i]=i;
	}
}
int ku(int k)
{
	int sum=0,i;
	for(i=0;i<k;i++)
	{
		int u=edge[i].u;
		int v=edge[i].v;
		int w=edge[i].w;
		int fa=find(u);
		int fb=find(v);
		if(fa!=fb||w<0)
		{
			sum+=w;
			pre[fa]=fb;
		}
	}
	return sum;
}
int main()
{
	while(scanf("%d%d",&n,&m)!=EOF)
	{
		int i,j;
		init(n);
		for(i=0;i<m;i++)
		{
			int u,v,w;
			scanf("%d%d%d",&u,&v,&w);
			edge[i].u=u;
			edge[i].v=v;
			edge[i].w=w;
		}
		k=m;
		for(i=1;i<=n;i++)
		{
			int cost;
			scanf("%d",&cost);
			if(cost!=-1)
			{
				edge[k].u=0;
				edge[k].v=i;
				edge[k++].w=cost;
			}
		}
		for(i=0;i<m;i++)
		{
			int fa=find(edge[i].u);
			int fb=find(edge[i].v);
			if(fa!=fb)
				pre[fa]=fb;
		}
		for(i=2;i<=n;i++)
		{
			if(find(1)!=find(i))
			{
				break;
			}
		}
		int temp;
		if(i==n+1)
		{
			init(n);
			qsort(edge,m,sizeof(edge[0]),cmp);
			temp=ku(m);
			init(n);
			qsort(edge,k,sizeof(edge[0]),cmp);
			int ans=ku(k);
			if(ans>temp)
				printf("%d\n",temp);
			else
				printf("%d\n",ans);
			continue;
		}
		init(n);
		qsort(edge,k,sizeof(edge[0]),cmp);
		int ans=ku(k);
		printf("%d\n",ans);
	}
}