結果
問題 | No.1507 Road Blocked |
ユーザー |
|
提出日時 | 2021-05-14 22:44:34 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 457 ms / 2,000 ms |
コード長 | 624 bytes |
コンパイル時間 | 100 ms |
コンパイル使用メモリ | 12,288 KB |
実行使用メモリ | 46,848 KB |
最終ジャッジ日時 | 2024-10-02 03:04:25 |
合計ジャッジ時間 | 11,609 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import sys sys.setrecursionlimit(10 ** 9) input = sys.stdin.buffer.readline N = int(input()) T = [[] for _ in range(N)] for _ in range(N - 1): u, v = (int(x) - 1 for x in input().split()) T[u].append(v) T[v].append(u) MOD = 998244353 modinv = lambda a, mod=MOD: pow(a, mod - 2, mod) ans = - (N - 1) * N // 2 def dfs(v, parent): global ans subt = 1 for x in T[v]: if x != parent: subt += dfs(x, v) ans += subt * (subt - 1) // 2 + (N - subt) * (N - subt - 1) // 2 ans %= MOD return subt dfs(0, -1) ans *= modinv((N - 1) * N * (N - 1) // 2) ans %= MOD print(ans)