結果
問題 | No.1507 Road Blocked |
ユーザー |
![]() |
提出日時 | 2021-05-14 23:55:37 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 583 ms / 2,000 ms |
コード長 | 782 bytes |
コンパイル時間 | 510 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 30,720 KB |
最終ジャッジ日時 | 2024-10-02 06:27:28 |
合計ジャッジ時間 | 18,704 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import sys input = sys.stdin.buffer.readline sys.setrecursionlimit(10 ** 7) N = int(input()) G = [[] for _ in range(N)] for _ in range(N-1): a, b = map(int, input().split()) a -= 1 b -= 1 G[a].append(b) G[b].append(a) par = [-1] * N que = [0] topo = [] while que: s = que.pop() topo.append(s) for t in G[s]: if t == par[s]: continue par[t] = s que.append(t) G[t].remove(s) ans = 0 mod = 998244353 size = [1] * N topo.reverse() choice = N * (N - 1) * (N - 1) choice_inv = pow(choice, mod-2, mod) for s in topo[:-1]: p = par[s] size[p] += size[s] x = size[s] y = N - size[s] ans += x * (x - 1) * choice_inv % mod ans += y * (y - 1) * choice_inv % mod ans %= mod print(ans)