結果
問題 | No.1507 Road Blocked |
ユーザー |
![]() |
提出日時 | 2021-05-14 22:44:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 354 ms / 2,000 ms |
コード長 | 640 bytes |
コンパイル時間 | 166 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 112,384 KB |
最終ジャッジ日時 | 2024-10-02 03:03:41 |
合計ジャッジ時間 | 11,548 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
mod = 998244353 n = int(input()) Adj = [set() for _ in range(n)] for _ in range(n-1): u, v = map(int, input().split()) Adj[u-1].add(v-1) Adj[v-1].add(u-1) Parent = [-1]*n Size = [0]*n D = [0] Sort = [] while D: v = D.pop() Sort.append(v) for nxt in Adj[v]: if nxt != Parent[v]: Parent[nxt] = v D.append(nxt) for i in Sort[::-1]: Size[i] += 1 for child in Adj[i]: if child != Parent[i]: Size[i] += Size[child] ans = 0 cff = pow((n*(n-1)**2)//2, mod-2, mod) for i in range(1, n): s = Size[i] ans += s*(n-s) ans %= mod print((mod+1-ans*cff)%mod)