結果

問題 No.1817 Reversed Edges
ユーザー CyanmondCyanmond
提出日時 2022-01-21 21:55:16
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 226 ms / 2,000 ms
コード長 2,871 bytes
コンパイル時間 2,024 ms
コンパイル使用メモリ 207,672 KB
実行使用メモリ 13,952 KB
最終ジャッジ日時 2024-05-04 12:45:06
合計ジャッジ時間 6,763 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 2 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 2 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 195 ms
9,216 KB
testcase_08 AC 85 ms
6,400 KB
testcase_09 AC 166 ms
8,832 KB
testcase_10 AC 95 ms
6,528 KB
testcase_11 AC 127 ms
7,552 KB
testcase_12 AC 226 ms
9,984 KB
testcase_13 AC 222 ms
9,984 KB
testcase_14 AC 202 ms
10,112 KB
testcase_15 AC 201 ms
10,112 KB
testcase_16 AC 208 ms
9,984 KB
testcase_17 AC 226 ms
9,984 KB
testcase_18 AC 204 ms
10,012 KB
testcase_19 AC 198 ms
9,984 KB
testcase_20 AC 207 ms
10,112 KB
testcase_21 AC 198 ms
9,984 KB
testcase_22 AC 187 ms
10,244 KB
testcase_23 AC 168 ms
10,368 KB
testcase_24 AC 192 ms
13,952 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#line 1 "paper.cpp"
#include <bits/stdc++.h>

#line 3 "library2/utility/int_alias.hpp"

using i8 = std::int8_t;
using u8 = std::uint8_t;
using i16 = std::int16_t;
using i32 = std::int32_t;
using i64 = std::int64_t;
using u16 = std::uint16_t;
using u32 = std::uint32_t;
using u64 = std::uint64_t;
#line 2 "library2/utility/rec_lambda.hpp"

template <class F> class RecursiveLambda {
    F f;

  public:
    explicit constexpr RecursiveLambda(F &&f_) : f(std::forward<F>(f_)) {}
    template <class... Args> constexpr auto operator()(Args &&...args) const {
        return f(*this, std::forward<Args>(args)...);
    }
};

template <class F> constexpr decltype(auto) rec_lambda(F &&f) {
    return RecursiveLambda<F>(std::forward<F>(f));
}
#line 3 "library2/utility/rep.hpp"

class Range {
    struct Iterator {
        int itr;
        constexpr Iterator(const int pos) noexcept : itr(pos) {}
        constexpr void operator++() noexcept {
            ++itr;
        }
        constexpr bool operator!=(const Iterator &other) const noexcept {
            return itr != other.itr;
        }
        constexpr int operator*() const noexcept {
            return itr;
        }
    };
    const Iterator first, last;

  public:
    explicit constexpr Range(const int f, const int l) noexcept
        : first(f), last(std::max(f, l)) {}
    constexpr Iterator begin() const noexcept {
        return first;
    }
    constexpr Iterator end() const noexcept {
        return last;
    }
};

constexpr Range rep(const int l, const int r) noexcept {
    return Range(l, r);
}
constexpr Range rep(const int n) noexcept {
    return Range(0, n);
}
#line 3 "library2/utility/scan.hpp"

template <typename T = int> T scan() {
    T ret;
    std::cin >> ret;
    return ret;
}
#line 7 "paper.cpp"

int main() {
    const int N = scan();
    std::vector<std::vector<int>> Tree(N);
    std::vector<int> A(N - 1), B(N - 1);
    for (const int i : rep(N - 1)) {
        A[i] = scan() - 1;
        B[i] = scan() - 1;
        Tree[A[i]].push_back(B[i]);
        Tree[B[i]].push_back(A[i]);
    }

    int revsum = 0;
    std::vector<int> prm(N);
    auto set = rec_lambda([&](auto &&f, const int v, const int p) -> void {
        for (const int to : Tree[v]) {
            if (to == p) {
                continue;
            }
            if (v > to) {
                ++revsum;
                --prm[to];
            } else {
                ++prm[to];
            }
            f(to, v);
        }
    });
    set(0, N);

    auto calc = rec_lambda([&](auto &&f, const int v, const int p) -> void {
        for (const int to : Tree[v]) {
            if (to == p) {
                continue;
            }
            prm[to] += prm[v];
            f(to, v);
        }
    });
    calc(0, N);

    for (const int i : rep(0, N)) {
        std::cout << revsum + prm[i] << std::endl;
    }
}
0