蓝桥学习 PREV-25

试题 历届试题

PREV-25 城市建设

问题描述

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

思路：

一道有点变形的最小生成树的题目，用到了最小生成树和并查集，首先我们先要判断一下题目给出的状态，虽然题目说明了给出的数据都有效，但是我们得先判断一下如果使用道路的时候的总值，因为如果直接做了码头的判断，可能因为只有一个合格码头而导致无法与其他道路正常连接，徒加工程费用。所以我们得先算出只铺设道路时的值，然后对比两个都建设的值，当然要注意的还有道路如果无法全部连接所有的城市，那其将无法使用，因为没有意义。
所有的港口为了能正常的连通，都得当成从0点出发到达该点的路线，只有经过这样的转换才可以正常的去计算。还有就是如果是消费为负的路就一定要加进来，因为是只赚不赔啊。

代码：

#include<iostream>
#include<algorithm>
using namespace std;
struct dian{
	int x,y,node;
}d[110100];
int cuan[10100],dp[10100],cnt=0;
bool cmp(dian q,dian p){
	return q.node<p.node;
}
int fun(int x){
	if(x!=dp[x]){
		return dp[x]=fun(dp[x]);
	}
	return x;
}
int fuck(int n,int m){
	int sum=0;
	for(int i=0;i<=n;i++)dp[i]=i;
	sort(d,d+m,cmp);
	for(int i=0;i<m;i++){
		int x=fun(d[i].x);
		int y=fun(d[i].y);
		if(dp[x]!=dp[y]''d[i].node<0){
			if(dp[x]!=dp[y])dp[x]=dp[y];
			sum+=d[i].node;
		}
	}
	return sum;
}
int main(){
	int n,m,flag=0;
	cin>>n>>m;
	for(int i=0;i<m;i++){
		cin>>d[i].x>>d[i].y>>d[i].node;
	}
	for(int i=1;i<=n;i++){
		cin>>cuan[i];
	}
	int num1=fuck(n,m);

	for(int i=1;i<=n;i++){
		if(dp[i]==i)flag++;
	}
	for(int i=1;i<=n;i++){
		if(cuan[i]!=-1){
			d[i+m-1].x=0;
			d[i+m-1].y=i;
			d[i+m-1].node=cuan[i];
		}
	}
	int num2=fuck(n,m+n);
	if(flag!=1)cout<<num2<<endl;
	else cout<<min(num1,num2)<<endl;
	return 0;
}