#656415 (C++17) No.1507 Road Blocked

• 問題: No.1507 Road Blocked
• ユーザー: oliverx3
• 提出日時: 2021-05-13 09:45:40
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 97 ms / 2,000 ms
• コード長: 1,158 bytes
• コンパイル時間: 4,190 ms
• コンパイル使用メモリ: 264,352 KB
• 実行使用メモリ: 17,152 KB
• 最終ジャッジ日時: 2024-09-24 23:17:08
• 合計ジャッジ時間: 8,302 ms
• ジャッジサーバーID （参考情報）: judge5 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	4 ms; 6,816 KB
testcase_01	AC	3 ms; 6,940 KB
testcase_02	AC	3 ms; 6,944 KB
testcase_03	AC	77 ms; 17,152 KB
testcase_04	AC	88 ms; 10,880 KB
testcase_05	AC	94 ms; 10,880 KB
testcase_06	AC	89 ms; 10,752 KB
testcase_07	AC	91 ms; 10,880 KB
testcase_08	AC	87 ms; 10,880 KB
testcase_09	AC	90 ms; 10,880 KB
testcase_10	AC	85 ms; 10,944 KB
testcase_11	AC	85 ms; 10,880 KB
testcase_12	AC	93 ms; 10,880 KB
testcase_13	AC	96 ms; 10,880 KB
testcase_14	AC	97 ms; 10,952 KB
testcase_15	AC	86 ms; 10,796 KB
testcase_16	AC	91 ms; 10,880 KB
testcase_17	AC	92 ms; 10,880 KB
testcase_18	AC	87 ms; 10,880 KB
testcase_19	AC	86 ms; 10,880 KB
testcase_20	AC	86 ms; 10,880 KB
testcase_21	AC	86 ms; 10,880 KB
testcase_22	AC	86 ms; 10,948 KB
testcase_23	AC	90 ms; 10,880 KB
testcase_24	AC	91 ms; 10,880 KB
testcase_25	AC	90 ms; 10,880 KB
testcase_26	AC	92 ms; 10,880 KB
testcase_27	AC	89 ms; 10,880 KB
testcase_28	AC	88 ms; 10,880 KB
testcase_29	AC	91 ms; 10,880 KB
testcase_30	AC	85 ms; 10,824 KB
testcase_31	AC	90 ms; 10,820 KB
testcase_32	AC	92 ms; 10,880 KB

ソースコード

#include <bits/stdc++.h>
#include <atcoder/all>

#define rep(i, n) for(int i = 0; i < (int)(n); i++)

using mint = atcoder::modint998244353;

mint dp[100010];

std::vector<int> G[100010];
bool seen[100010];

int dfs(int node = 0) {
    for(const auto &to:G[node]) {
        if(seen[to]) continue;
        seen[to] = true;
        dp[node] += dfs(to);
    }
    return (++dp[node]).val();
}

int main() {
    long long n;
    std::cin >> n;
    std::vector<std::pair<long long, long long>> v(n-1);
    rep(i, n-1) {
        int a, b;
        std::cin >> a >> b;
        a--, b--;
        v[i] = {a, b};
        G[a].push_back(b);
        G[b].push_back(a);
    }
    seen[0] = true;
    dfs();
    //rep(i, n) std::cout << dp[i].val() << ' ';std::endl(std::cout);
    mint ans = 0;
    for(auto [a, b]:v) {
        if(dp[a].val()<dp[b].val()) std::swap(a, b);
        ans += (n-dp[b])*dp[b];
        //std::cout << (dp[a]-dp[b]).val() << ' ' << dp[b].val() << std::endl;
    }
    //rep(i, n) ans += dp[i]*(n-dp[i]);
    //std::cout << ans.val() << std::endl;
    ans/=(n-1)*n*(n-1)/2;
    ans = 1-ans;
    std::cout << ans.val() << std::endl;
    return 0;
}