蓝桥杯历届试题-城市建设 （C++代码）

解题思路:
    首先了解并查集，然后使用克鲁斯卡尔算法找到最小生成树。
参考代码:

#include<cstdio>
#include<algorithm>
#define maxn 10010
#define maxm 100010
#define N 10000000
using namespace std;
int n,m,pre[maxn]; 
struct edge{
	int begin,end,s;
}Edge[maxm+maxn];
int cmp(struct edge x,struct edge y){ return x.s < y.s; }
int mymin(int a,int b){ return a<b?a:b; }
int find(int r)
{
	int t = r;
	while(pre[r]!=r) r=pre[r];
	while(t!=r)
	{
		int temp=t;
		pre[t]=r;
		t=pre[temp];
	}
	return r;
}
int main(void)
{
	scanf("%d%d",&n,&m);
	for(int i=0;i<m;i++) scanf("%d%d%d",&Edge[i].begin,&Edge[i].end,&Edge[i].s);
	for(int i=m;i<m+n;i++)
	{	
		Edge[i].begin=0; Edge[i].end=i-m+1;
		scanf("%d",&Edge[i].s);
		if(Edge[i].s==-1) Edge[i].s=-N;
	 } 
	 sort(Edge,Edge+m+n,cmp);
	 int sum1=0,sum2=0,cnt=0;
	 for(int i=0;i<=n;i++) pre[i]=i;
	 for(int i=0;i<m+n;i++)
	 {
	 	if(Edge[i].s == -N) continue;
	 	int x = find(Edge[i].begin);
	 	int y = find(Edge[i].end);
	 	if(x!=y)
	 	{
	 		pre[x]=y;
	 		sum1+=Edge[i].s;
		 }
		 else if(Edge[i].s<0) sum1+=Edge[i].s;
	 }
	 for(int i=1;i<=n;i++) pre[i]=i;
	 for(int i=0;i<n+m;i++)
	 {
	 	if(Edge[i].begin==0) continue;
	 	int x = find(Edge[i].begin);
	 	int y = find(Edge[i].end);
	 	if(x!=y) 
	 	{
	 		pre[x]=y;
			sum2+=Edge[i].s;
		 }
		 else if(Edge[i].s<0) sum2+=Edge[i].s;
	 }
	 for(int i=1;i<=n;i++) if(pre[i]==i) cnt++;
	 if(cnt!=1) printf("%d",sum1);
	 else printf("%d",mymin(sum1,sum2));
	return 0;
}