题目描述:
栋栋居住在一个繁华的C市中,然而,这个城市的道路大都年久失修。
市长准备重新修一些路以方便市民,于是找到了栋栋,希望栋栋能帮助他。
C市中有n个比较重要的地点,市长希望这些地点重点被考虑。
现在可以修一些道路来连接其中的一些地点,每条道路可以连接其中的两个地点。
另外由于C市有一条河从中穿过,也可以在其中的一些地点建设码头,所有建了码头的地点可以通过河道连接。
栋栋拿到了允许建设的道路的信息,
包括每条可以建设的道路的花费,以及哪些地点可以建设码头和建设码头的花费。
市长希望栋栋给出一个方案,使得任意两个地点能只通过新修的路或者河道互达,同时花费尽量小。
样例说明
建设第2、3、4条道路,在地点4、5建设码头,总的花费为9。
数据规模和约定
对于100%的数据,1 < = n < = 10000,
1 < = m < = 100000,-1000< =c< =1000,
-1< =w_i< =1000,
w_i≠0。
输入:
输入的第一行包含两个整数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
参考解答:
import java.util.Scanner;
import java.util.Arrays;
import java.util.Comparator;
class edge{
int island_a;
int island_b;
int weight;
public edge(int a, int b, int weight) {
this.island_a = a;
this.island_b = b;
this.weight = weight;
}
}
public class Main {
private static int[] tree;
private static edge[] edges;
private static int num_island;
private static int num_edges;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
num_island = sc.nextInt();
num_edges = sc.nextInt();
tree = new int[num_island + 2];
edges = new edge[3 * num_edges];
for (int i = 1; i <= num_edges; i++) {
edges[i] = new edge(sc.nextInt(), sc.nextInt(), sc.nextInt());
}
int no_terminal = kruskal();
num_island++;
for (int i = 1; i < num_island; i++) {
int weight = sc.nextInt();
if (weight != -1) {
num_edges++;
edges[num_edges] = new edge(i, num_island, weight);
}
}
int establish_terminal = kruskal();
System.out.println(Math.min(no_terminal,establish_terminal));
}
private static int find_root(int node) {
if (tree[node] != node) {
return tree[node] = find_root(tree[node]);
}
return node;
}
private static boolean merge(int island_a, int island_b, int weight) {
int root_a = find_root(island_a);
int root_b = find_root(island_b);
if (root_a != root_b || weight < 0) {
tree[root_b] = root_a;
return true;
}
return false;
}
private static void init() {
for (int i = 1; i <= num_island; i++) {
tree[i] = i;
}
}
private static int kruskal() {
int sum = 0;
init();
Arrays.sort(edges, 1, num_edges + 1, new Comparator<edge>() {
public int compare(edge a, edge b) {
return a.weight - b.weight;
}
});
int count = 0;
for (int i = 1; i <= num_edges; i++) {
if (merge (edges[i].island_a, edges[i].island_b, edges[i].weight)) {
sum += edges[i].weight;
count++;
}
}
if (count < num_island - 1) {
return Integer.MAX_VALUE;
}
return sum;
}
}