結果
問題 | No.1507 Road Blocked |
ユーザー |
![]() |
提出日時 | 2021-05-14 22:23:09 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 492 ms / 2,000 ms |
コード長 | 779 bytes |
コンパイル時間 | 194 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 96,512 KB |
最終ジャッジ日時 | 2024-10-02 01:57:28 |
合計ジャッジ時間 | 15,148 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
def tree_dfs(g, root=0): s = [root] d = [1] * len(g) order = [] while s: p = s.pop() d[p] = 0 order.append(p) for node in g[p]: if d[node]: s.append(node) return order MOD = 998244353 n = int(input()) g = [[] for _ in range(n)] for i in range(n - 1): a, b = map(int, input().split()) g[a - 1].append(b - 1) g[b - 1].append(a - 1) order = tree_dfs(g) d = [0] * (n + 1) for v in order[::-1]: for node in g[v]: d[v] += d[node] d[v] += 1 ans = 0 for v in range(1, n): ans += d[v] * (d[v] - 1) * pow(2, MOD - 2, MOD) + (n - d[v]) * (n - d[v] - 1) * pow(2, MOD - 2, MOD) print((ans * pow(n * (n - 1) * pow(2, MOD - 2, MOD), MOD - 2, MOD) * pow(n - 1, MOD - 2, MOD)) % MOD)