結果

問題 No.806 木を道に
ユーザー mencottonmencotton
提出日時 2020-11-02 22:09:26
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,370 bytes
コンパイル時間 874 ms
コンパイル使用メモリ 80,256 KB
実行使用メモリ 14,380 KB
最終ジャッジ日時 2023-09-29 13:51:35
合計ジャッジ時間 3,155 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,380 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 AC 2 ms
4,380 KB
testcase_05 WA -
testcase_06 WA -
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 1 ms
4,376 KB
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 AC 40 ms
10,128 KB
testcase_25 AC 60 ms
14,380 KB
testcase_26 AC 20 ms
5,124 KB
testcase_27 AC 64 ms
10,468 KB
testcase_28 AC 3 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

template<typename T = int>
struct Edge {
    int from, to;
    T cost;
    int idx;

    Edge() = default;

    Edge(int from, int to, T cost = 1, int idx = -1) : from(from), to(to), cost(cost), idx(idx) {}

    operator int() const { return to; }
};

template<typename T = int>
struct Graph {
    vector<vector<Edge<T>>> g;
    int es;

    Graph() = default;

    explicit Graph(int n) : g(n), es(0) {}

    size_t size() const {
        return g.size();
    }

    void add_directed_edge(int from, int to, T cost = 1) {
        g[from].emplace_back(from, to, cost, es++);
    }

    void add_edge(int from, int to, T cost = 1) {
        g[from].emplace_back(from, to, cost, es);
        g[to].emplace_back(to, from, cost, es++);
    }

    void read(int M, int padding = -1, bool weighted = false, bool directed = false) {
        for (int i = 0; i < M; i++) {
            int a, b;
            cin >> a >> b;
            a += padding;
            b += padding;
            T c = T(1);
            if (weighted) cin >> c;
            if (directed) add_directed_edge(a, b, c);
            else add_edge(a, b, c);
        }
    }
};

template<typename T = int>
struct TreeDiameter : Graph<T> {
public:
    using Graph<T>::Graph;
    using Graph<T>::g;
    vector<Edge<T>>
            path;

    T build() {
        to.assign(g.size(), -1);
        auto p = dfs(0, -1);
        auto q = dfs(p.second, -1);

        int now = p.second;
        while (now != q.second) {
            for (auto &e : g[now]) {
                if (to[now] == e.to) {
                    path.emplace_back(e);
                }
            }
            now = to[now];
        }
        return q.first;
    }

    explicit TreeDiameter(const Graph<T> &g) : Graph<T>(g) {}

private:
    vector<int> to;

    pair<T, int> dfs(int idx, int par) {
        pair<T, int> ret(0, idx);
        for (auto &e : g[idx]) {
            if (e.to == par) continue;
            auto cost = dfs(e.to, idx);
            cost.first += e.cost;
            if (ret < cost) {
                ret = cost;
                to[idx] = e.to;
            }
        }
        return ret;
    }
};

int main() {
    int n;
    cin >> n;
    TreeDiameter<int> g(n);
    g.read(n - 1, -1);
    cout << n - 1 - g.build() << endl;
    return 0;
}
0