历届试题 城市建设
时间限制: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。
思路:这道看,想了一想,最小生成树啊!(注意目的是任意两个地点互达)。但是问题来了,这里不仅有道路,
还有码头,码头就比较棘手了,看起来是点的权值。那应该怎么办?
把点转换为边就行了!多附加一个点,不妨设编号为0,然后两者连一条边,将点的权值改成该条边的权值。
这时候,问题就差不多变成了求0~n的最小生成树问题。(注意:如果w_i为-1,则表示地点i无法建设码头,
此时就没有必要连边了)
注意,有可能没有建码头而仅仅建立道路使1~n任意两点互达,这样做的花费最小。而此时,0就不会与
1~n相连了。所以该情况要先特殊考虑!
然后题目还有个陷阱:花费为负数!所以最小生成树用Kruskal算法做完以后,还要再把负边给加完。
但是还是会超时,怎么办?并查集加个rank,然后循环添边的时候,如果任意两点已经互通(可通过边的
数目判断)且当前边权值已经非负,则直接跳出循环。
代码如下:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;
import java.util.StringTokenizer;
class Reader {
static BufferedReader reader;
static StringTokenizer tokenizer;
static void init(InputStream input) {
reader = new BufferedReader(new InputStreamReader(input));
tokenizer = new StringTokenizer("");
}
static String next() throws IOException {
while (!tokenizer.hasMoreTokens()) {
tokenizer = new StringTokenizer(reader.readLine());
}
return tokenizer.nextToken();
}
static int nextInt() throws IOException {
return Integer.parseInt(next());
}
}
class Edge {
int u, v, value;
/**
* @param u
* @param v
* @param value
*/
public Edge(int u, int v, int value) {
super();
this.u = u;
this.v = v;
this.value = value;
}
/**
*
*/
public Edge() {
super();
// TODO Auto-generated constructor stub
}
}
public class Main {
static int pa[], rank[], n, m, size;
static Edge arr[];
static Comparator<Edge> cmp = new Comparator<Edge>() {
@Override
public int compare(Edge arg0, Edge arg1) {
// TODO Auto-generated method stub
return (arg0.value - arg1.value);
}
};
private static int parent(int n) {
if (pa[n] == n)
return n;
pa[n] = parent(pa[n]);
return pa[n];
}
private static int getmin(int a, int b) {
return a < b ? a : b;
}
public static void main(String[] args) throws IOException {
// TODO Auto-generated method stub
Reader.init(System.in);
n = Reader.nextInt();
m = Reader.nextInt();
arr = new Edge[m + n + 1];
pa = new int[n + 1];
rank = new int[n + 1];
for (int i = 1; i <= m; i++) {
arr[i] = new Edge();
arr[i].u = Reader.nextInt();
arr[i].v = Reader.nextInt();
arr[i].value = Reader.nextInt();
}
size = m;
int num;
for (int i = 1; i <= n; i++) {
num = Reader.nextInt();
if (num != -1)
arr[++size] = new Edge(0, i, num);
}
for (int i = 0; i <= n; i++) {
pa[i] = i;
rank[i] = 1;
}
if (n == 1)
System.out.println(0);
else {
Arrays.sort(arr, 1, m + 1, cmp);
int ans = Integer.MAX_VALUE;
int sum = 0;
int total = 0;
for (int i = 1; i <= m; i++) {
int pau = parent(arr[i].u);
int pav = parent(arr[i].v);
if (pau != pav) {
sum += arr[i].value;
total++;
if (rank[pau] <= rank[pav])
pa[pau] = pav;
else
pa[pav] = pau;
if (rank[pau] == rank[pav])
rank[pav]++;
} else if (arr[i].value < 0)
sum += arr[i].value;
if ((total == n - 1) && (arr[i].value >= 0))
break;
}
if (total == n - 1)
ans = getmin(ans, sum);
Arrays.sort(arr, 1, size + 1, cmp);
total = 0;
sum = 0;
for (int i = 0; i <= n; i++) {
pa[i] = i;
rank[i] = 1;
}
for (int i = 1; i <= size; i++) {
int pau = parent(arr[i].u);
int pav = parent(arr[i].v);
if (pau != pav) {
sum += arr[i].value;
total++;
if (rank[pau] <= rank[pav])
pa[pau] = pav;
else
pa[pav] = pau;
if (rank[pau] == rank[pav])
rank[pav]++;
} else if (arr[i].value < 0)
sum += arr[i].value;
if ((total == n) && (arr[i].value >= 0))
break;
}
if (total == n)
ans = getmin(ans, sum);
System.out.println(ans);
}
}
}