結果

問題 No.1639 最小通信路
ユーザー magsta
提出日時 2021-07-09 15:48:51
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 32 ms / 2,000 ms
コード長 1,958 bytes
コンパイル時間 7,727 ms
コンパイル使用メモリ 303,128 KB
実行使用メモリ 8,976 KB
最終ジャッジ日時 2024-09-17 01:00:02
合計ジャッジ時間 9,590 ms
ジャッジサーバーID
(参考情報)
judge6 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 43
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include "testlib.h"
using namespace std;

struct UnionFind {
    vector<int> size, parents;
    UnionFind() {}
    UnionFind(int n) {
        size.resize(n, 0);
        parents.resize(n, 0);
        for (int i = 0; i < n; i++) {
            makeTree(i);
        }
    }
    void makeTree(int x) {
        parents[x] = x;
        size[x] = 1;
    }
    bool isSame(int x, int y) {
        return findRoot(x) == findRoot(y);
    }
    bool unite(int x, int y) {
        x = findRoot(x);
        y = findRoot(y);
        if (x == y) return false;
        if (size[x] > size[y]) {
            parents[y] = x;
            size[x] += size[y];
        }
        else {
            parents[x] = y;
            size[y] += size[x];
        }
        return true;
    }
    int findRoot(int x) {
        if (x != parents[x]) {
            parents[x] = findRoot(parents[x]);
        }
        return parents[x];
    }
    int treeSize(int x) {
        return size[findRoot(x)];
    }
};

int main() {
    registerValidation();
    int N = inf.readInt(2, 100000, "N");
    inf.readEoln();
    UnionFind uni(N);
    int count = 0;
    map<pair<int, int>, int> Map;
    bool flag = true;
    for (int i = 0; i < N * (N - 1) / 2; i++) {
        int a = inf.readInt(1, N, "a");
        inf.readSpace();
        int b = inf.readInt(1, N, "a");
        inf.readSpace();
        string S = inf.readToken("[0-9]{1, 101}", "S");
        inf.readEoln();
        if (Map[make_pair(a, b)] == 1) cout << -1 << endl;
        Map[make_pair(a, b)]++;
        a--; b--;
        if (S.size() == 101) {
            for (int i = 2; i < 101; i++) {
                if (S[i] != '0') cout << -1 << endl;
            }
        }
        if (!uni.isSame(a, b)) {
            uni.unite(a, b);
            count++;
            if (count == N - 1 && flag) {
                cout << S << endl;
                flag = false;
            }
        }
    }
    inf.readEof();
}
0