蓝桥杯--城市建设

题目链接

题目大概意思就是说有n个点 m条边。如果只是这样的话就是最小生成树的裸体 (裸题)。但是他还有一个东西。码头。就是有些地方可以建码头,建码头的各个点之间可以相互到达。问求最小的花费。注意可以是负数这个条件。

思路：就是最小生成树的变形题了，因为有的地方不可以建码头，我们可以虚拟出来一个点0,可以建码头的点就相当于和0建立了一条边，权值为建码头所需要的费用。这样的话我们就可以按照最小生成树去做了。注意我们要建立所有的负权道路,不管他俩联通与否。因为建立这条道路可以赚钱。

坑点：最后有一个坑点。就是如果与0相连的边只用了一条。我们要删除建立这个码头的权值。为什么呢？举个栗子。

5 10
5 2 8
4 5 8
2 5 10
1 4 4
5 3 5
1 2 9
1 5 8
5 1 2
1 3 5
4 3 12
-1 -1 -1 18 11

如果按照上面的思路做的话，求出来的应该是 30。但是答案是 19 。为什么呢？因为最后一条边没必要建立，最后一条建立的边是联通0的。也就是说只建立了一个码头。那这个码头并不能到达剩下n-1城市中的任意一个城市。也就是说建立这个码头并没有带来任何收益。至少建立两个码头才能使修建道路的费用变少。所以我们要特判建立一个码头的情况就ok了。

代码：  

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+10;
#define inf 0x3f3f3f3f
int mt[maxn];
struct node{
	int u,v,w;
};
vector<node>v;
int n,m;
int fa[maxn];
int get(int x)
{
	if(x == fa[x])
		return x;
	return fa[x] = get(fa[x]);
}
int Union(int x,int y)
{
	fa[get(x)] = fa[get(y)];
}
bool cmp(node a,node b)
{
	return a.w < b.w;
}
int main()
{
	cin>>n>>m;
	node temp;
	for(int i=1;i<=n;i++)
		fa[i] = i;
	for(int i=1;i<=m;i++)
	{
		cin>>temp.u>>temp.v>>temp.w;
		v.push_back(temp);		
	}
	for(int i=1;i<=n;i++)
	{
		int w;
		cin>>w;
		if(w == -1)continue;
		temp.u = 0;
		temp.v = i;
		temp.w = w;
		v.push_back(temp);
	}
	sort(v.begin(),v.end(),cmp);
	int len = v.size();
	int cost = 0;
	int cnt = 0;
	int zero = 0;
	for(int i=0;i<len;i++)
	{
		int a = v[i].u;
		int b = v[i].v;
		if(get(a) == get(b) && v[i].w > 0)continue;
		if(a == 0)
			zero++;
		Union(a,b); 
		cost += v[i].w;
	}
	if(zero == 1)
	{
		for(int i=0;i<len;i++)
		{
			if(v[i].u == 0)
			{
				cost -= v[i].w;
				break;
			}
		}	
	}
	cout<<cost<<endl;
	return 0;
}

/*
5 10
5 2 8
4 5 8
2 5 10
1 4 4
5 3 5
1 2 9
1 5 8
5 1 2
1 3 5
4 3 12
-1 -1 -1 18 11 
*/