結果
問題 | No.1507 Road Blocked |
ユーザー | qib |
提出日時 | 2022-10-20 17:28:18 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 291 ms / 2,000 ms |
コード長 | 933 bytes |
コンパイル時間 | 315 ms |
コンパイル使用メモリ | 87,084 KB |
実行使用メモリ | 100,740 KB |
最終ジャッジ日時 | 2023-09-12 22:09:35 |
合計ジャッジ時間 | 14,129 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 71 ms
71,356 KB |
testcase_01 | AC | 71 ms
71,092 KB |
testcase_02 | AC | 71 ms
71,256 KB |
testcase_03 | AC | 200 ms
100,740 KB |
testcase_04 | AC | 277 ms
94,432 KB |
testcase_05 | AC | 277 ms
94,320 KB |
testcase_06 | AC | 282 ms
94,260 KB |
testcase_07 | AC | 278 ms
94,128 KB |
testcase_08 | AC | 282 ms
94,456 KB |
testcase_09 | AC | 290 ms
95,072 KB |
testcase_10 | AC | 279 ms
94,428 KB |
testcase_11 | AC | 280 ms
94,464 KB |
testcase_12 | AC | 281 ms
94,196 KB |
testcase_13 | AC | 284 ms
94,184 KB |
testcase_14 | AC | 281 ms
94,552 KB |
testcase_15 | AC | 276 ms
94,320 KB |
testcase_16 | AC | 286 ms
94,400 KB |
testcase_17 | AC | 281 ms
94,212 KB |
testcase_18 | AC | 280 ms
94,224 KB |
testcase_19 | AC | 285 ms
94,460 KB |
testcase_20 | AC | 291 ms
94,488 KB |
testcase_21 | AC | 285 ms
94,300 KB |
testcase_22 | AC | 279 ms
94,188 KB |
testcase_23 | AC | 281 ms
94,352 KB |
testcase_24 | AC | 278 ms
94,644 KB |
testcase_25 | AC | 276 ms
94,588 KB |
testcase_26 | AC | 282 ms
94,308 KB |
testcase_27 | AC | 283 ms
94,472 KB |
testcase_28 | AC | 279 ms
94,444 KB |
testcase_29 | AC | 275 ms
94,328 KB |
testcase_30 | AC | 279 ms
94,320 KB |
testcase_31 | AC | 275 ms
94,436 KB |
testcase_32 | AC | 270 ms
94,632 KB |
ソースコード
MOD = 998244353 n = int(input()) g = [[] for _ in range(n)] edges = [] for _ in range(n - 1): u, v = map(int, input().split()) g[u - 1].append(v - 1) g[v - 1].append(u - 1) edges.append((u - 1, v - 1)) dp = [0 for _ in range(n)] src = 0 par = [None for _ in range(n)] st = [~ src, src] while len(st) > 0: cur = st.pop() if cur >= 0: for nxt in g[cur]: if nxt == par[cur]: continue par[nxt] = cur st.append(~ nxt) st.append(nxt) else: dp[~ cur] += 1 if not par[~ cur] is None: dp[par[~ cur]] += dp[~ cur] ans = 0 for u, v in edges: if par[u] == v: ans = (ans + dp[u] * (dp[u] - 1) // 2) % MOD ans = (ans + (n - dp[u]) * (n - dp[u] - 1) // 2) % MOD elif par[v] == u: ans = (ans + dp[v] * (dp[v] - 1) // 2) % MOD ans = (ans + (n - dp[v]) * (n - dp[v] - 1) // 2) % MOD ans = (ans * pow(n - 1, MOD - 2, MOD)) % MOD ans = (ans * pow(n * (n - 1) // 2, MOD - 2, MOD)) % MOD print(ans)