历届试题 城市建设

@ 蓝桥杯 练习系统 历届试题 PREV-25

资源限制

时间限制：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无法建设码头。
输入保证至少存在一个方法使得任意两个地点能只通过新修的路或者河道互达。

---

输出格式

输出一行，包含一个整数，表示使得所有地点通过新修道路或者码头连接的最小花费。如果满足条件的情况下还能赚钱，那么你应该输出一个负数。

---

测试样例₁

Input：
5 5
1 2 4
1 3 -1
2 3 3
2 4 5
4 5 10
-1 10 10 1 1

Output：
9

Explanation：
建设第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。

---

ACcode：

import java.io.IOException;
import java.io.InputStream;
import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.*;

public class Main {

    static int n;
    static List<Edge>[] graph;

    public static void main(String[] args) {
        InputReader in = new InputReader(System.in, " ");
        n = in.nextInt(); int m = in.nextInt(), cnt = 0;
        graph = new List[n + 1];
        for (int i = 0; i <= n; i++)
            graph[i] = new ArrayList();
        { int v, w, value; while (m-- > 0) {
            v = in.nextInt();
            w = in.nextInt();
            value = in.nextInt();
            if (value < 0) {
                cnt += value;
                value = 0;
            }
            graph[v].add(new Edge(w, value));
            graph[w].add(new Edge(v, value));
        }}
        for (int i = 1, v; i <= n; i++) {
            v = in.nextInt();
            if (v == -1) continue;
            graph[0].add(new Edge(i, v));
            graph[i].add(new Edge(0, v));
        }
        System.out.print(Math.min(Tree(cnt, true), Tree(cnt, false)));
    }

    static int Tree(int cnt, boolean zero) {
        Queue<Edge> queue = new PriorityQueue();
        boolean[] marked = new boolean[n + 1];
        marked[0] = zero;
        queue.add(new Edge(1, 0));
        while (queue.size() > 0) {
            Edge edge = queue.poll();
            if (marked[edge.w]) continue;
            cnt += edge.value;
            marked[edge.w] = true;
            for (Edge e: graph[edge.w]) {
                if (marked[e.w]) continue;
                queue.add(e);
            }
        }
        if (zero) for (int i = 1; i <= n; i++) if (!marked[i]) return Integer.MAX_VALUE;
        return cnt;
    }

    static class Edge implements Comparable<Edge> {

        int w, value;

        Edge(int w, int value) {
            this.w = w;
            this.value = value;
        }

        public int compareTo(Edge edge) {
            return this.value - edge.value;
        }
    }

    static class InputReader {

        BufferedReader read;
        StringTokenizer tok;
        String delimiters;

        InputReader(InputStream in) { this(in, " \n\t\r\f"); }

        InputReader(InputStream in, String delimiters) {
            this.read = new BufferedReader(new InputStreamReader(in));
            this.tok = new StringTokenizer("", this.delimiters = delimiters);
        }

        String next() {
            while (!tok.hasMoreTokens())
                try {
                    tok = new StringTokenizer(read.readLine(), delimiters);
                } catch (IOException e) { }
            return tok.nextToken();
        }

        int nextInt() { return Integer.parseInt(next()); }
    }
}

因为计算 比较 码头的连通很麻烦，所以跑了两遍最小树生成