HDU1879 继续畅通工程（kruskal）

Problem

至少把kruskal练的很熟了已经。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;

class Main {
    static BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    static int N = 110, M = N * (N - 1), n, m, cnt, T, k;
    static int res;
    static int p[];
    static Pair pair[];

    public static void init() {
        p = new int[N];
        pair = new Pair[M];
        for (int i = 0; i < N; i++) p[i] = i;
        res = 0;
        cnt = 0;
        k = 0;
    }

    public static void main(String[] args) throws IOException {

        while (true) {
            n = Integer.parseInt(br.readLine());
            if (n == 0) break;
            init();
            k = n * (n - 1) >> 1;
            int idx = 0;
            for (int i = 0; i < k; i++) {
                String s[] = br.readLine().split(" ");
                int a = Integer.parseInt(s[0]);
                int b = Integer.parseInt(s[1]);
                int w = Integer.parseInt(s[2]);
                int st = Integer.parseInt(s[3]);
                if (st == 1) {
                    a = find(a);
                    b = find(b);
                    if (a != b) p[a] = b;
                } else pair[idx++] = new Pair(a, b, w);
            }

            Arrays.sort(pair, 0, idx);

            for (int i = 0; i < idx; i++) {
                int a = find(pair[i].a);
                int b = find(pair[i].b);
                int w = pair[i].w;
                if (a != b) {
                    p[a] = b;
                    res += w;
                    cnt++;
                }
            }

            System.out.println(res);

        }

    }

    public static int find(int x) {
        if (p[x] != x) p[x] = find(p[x]);
        return p[x];
    }

}

class Pair implements Comparable<Pair> {
    int a, b, w;

    public Pair(int a, int b, int w) {
        this.a = a;
        this.b = b;
        this.w = w;
    }

    @Override
    public int compareTo(Pair pair) {
        return this.w - pair.w;
    }
}