csp201412_4(最优灌溉)(Java100分)

问题描述
　　雷雷承包了很多片麦田，为了灌溉这些麦田，雷雷在第一个麦田挖了一口很深的水井，所有的麦田都从这口井来引水灌溉。
　　为了灌溉，雷雷需要建立一些水渠，以连接水井和麦田，雷雷也可以利用部分麦田作为“中转站”，利用水渠连接不同的麦田，这样只要一片麦田能被灌溉，则与其连接的麦田也能被灌溉。
　　现在雷雷知道哪些麦田之间可以建设水渠和建设每个水渠所需要的费用（注意不是所有麦田之间都可以建立水渠）。请问灌溉所有麦田最少需要多少费用来修建水渠。
输入格式
　　输入的第一行包含两个正整数n, m，分别表示麦田的片数和雷雷可以建立的水渠的数量。麦田使用1, 2, 3, ……依次标号。
　　接下来m行，每行包含三个整数ai, bi, ci，表示第ai片麦田与第bi片麦田之间可以建立一条水渠，所需要的费用为ci。
输出格式
　　输出一行，包含一个整数，表示灌溉所有麦田所需要的最小费用。
样例输入
4 4
1 2 1
2 3 4
2 4 2
3 4 3
样例输出
6
样例说明
　　建立以下三条水渠：麦田1与麦田2、麦田2与麦田4、麦田4与麦田3。
评测用例规模与约定
　　前20%的评测用例满足：n≤5。
　　前40%的评测用例满足：n≤20。
　　前60%的评测用例满足：n≤100。
　　所有评测用例都满足：1≤n≤1000，1≤m≤100,000，1≤ci≤10,000。

思路：这个题是最简单的关于最小生成树的题，我用的是prim算法（即每次找最短边，判断是否联通）判断是否联通用了并查集。

很基础的最小生成树，自行百度吧。我就不详解了。

import java.util.PriorityQueue;
import java.util.Scanner;

public class Main{
    static int pre[];

    static boolean isLinked(int a, int b) {
        int xx = find(a);
        int yy = find(b);
        if (xx == yy) {
            return true;
        } else {
            return false;
        }
    }

    static void Link(int a, int b) {
        int xx = find(a);
        int yy = find(b);
        if (xx > yy) {
            pre[xx] = yy;
        } else {
            pre[yy] = xx;
        }
    }

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

    public static void main(String[] args) {
        // TODO Auto-generated method stub
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int m = sc.nextInt();
        pre = new int[n + 1];
        for (int i = 0; i < pre.length; i++) {
            pre[i] = i;
        }
        PriorityQueue<Canal> queue = new PriorityQueue<Canal>();
        for (int i = 0; i < m; i++) {
            queue.add(new Canal(sc.nextInt(), sc.nextInt(), sc.nextInt()));
        }
        int result = 0;
        while (!queue.isEmpty()) {
            Canal c = queue.poll();
            if (!isLinked(c.start, c.end)) {
                Link(c.start, c.end);
                result += c.price;
            }
        }
        System.out.println(result);
    }
}

class Canal implements Comparable<Canal> {// 水渠
    int start;
    int end;
    int price;

    public Canal(int start, int end, int price) {
        super();
        this.start = start;
        this.end = end;
        this.price = price;
    }

    @Override
    public String toString() {
        return "Canal [start=" + start + ", end=" + end + ", price=" + price + "]";
    }

    @Override
    public int compareTo(Canal o) {
        // TODO Auto-generated method stub
        return price - o.price;
    }
}

欢迎指正！！！

csp201412_4(最优灌溉)(Java100分)

猜你喜欢