历届试题 城市建设

  历届试题 城市建设  

时间限制：1.0s   内存限制：256.0MB

    

问题描述

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

 

最开始想要直接通过kruskal做题，将码头问题通过两层for循环来算法两个点有通路的情况下的消耗，显然被wa了。

思考了一下感觉方法没错，那必然是错在转化上面了

两个码头间的花费确实是两个码头的花费，但是如果一个码头同时连接着其他两个码头，那么花费就是重复。

find了一下题解

发现了解决这个问题的办法，就是假想一个点0，所以可建造码头的地点都与其相连从而解决问题

其次需要注意的是，码头有且只有存在两个码头是才能发挥作用。故只有一个被征用的时候只能说明0点被当作城市加入了城市建设中，导致了没必要的花费。这种情况下，就可以发现没有码头发挥作用，那么就计算没有码头时的最小花费就好，也就是最小生成树。

还有就是此题没必要找边长的个数，因为存在赚钱的边是一定要加入并查集的

#include <bits/stdc++.h>
#define MAX 1001
using namespace std;

struct node
{
	int u,e,w;
}edge[900010];

int flag = 0;
int book[100100];
int pre[100100];

bool cmp(node a, node b)
{
	return a.w < b.w;
}

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

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

int kruskal_first(int n, int m)//有码头参与 
{
	init(n);
	int sum = 0;
	flag = 0;
	for(int i = 0; i < m; i++)
	{
		if(edge[i].w == MAX) continue;
		int t1 = find(edge[i].u);
		int t2 = find(edge[i].e);
		if(t1!=t2 || edge[i].w < 0)
		{
			pre[t1] = t2;
			sum += edge[i].w; 
			
			if(edge[i].u == 0) flag++;
		}
	}
	return sum;
}

int kruskal_second(int n, int m)
{
	init(n);
	int sum = 0;
	for(int i = 0; i < m; i++)
	{
		if(edge[i].u == 0) continue;
		int t1 = find(edge[i].u);
		int t2 = find(edge[i].e);
		if(t1!=t2 || edge[i].w < 0)
		{
			pre[t1] = t2;
			sum += edge[i].w; 
		}
	}
	return sum;
}

int main()
{
	int n, m;
	scanf("%d %d", &n, &m);
	for(int i = 0 ; i < m; i++)
	{
		scanf("%d %d %d", &edge[i].u, &edge[i].e, &edge[i].w);
	}
	for(int i = 1; i <= n; i++)
	{
		scanf("%d", &book[i]);
	}
	for(int i = 1; i <= n; i++)
	{
		edge[m].u = 0; edge[m].e = i;
		if(book[i] == -1) edge[m].w = MAX;
		else edge[m].w = book[i];
		m++;
	}
	
	sort(edge, edge+m, cmp);
	
	int ans1 = kruskal_first(n, m);
	//cout << flag << endl;
	if(flag == 1)
	{
		int ans2 = kruskal_second(n, m);
		printf("%d\n", ans2);
	}
	else
	{
		printf("%d\n", ans1);
	}
	//cout << ans1 << " " << ans2 << endl;
	
	
	return 0;
 }