結果
問題 | No.1507 Road Blocked |
ユーザー |
|
提出日時 | 2023-01-15 19:43:02 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 526 ms / 2,000 ms |
コード長 | 924 bytes |
コンパイル時間 | 187 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 46,848 KB |
最終ジャッジ日時 | 2024-12-29 10:53:27 |
合計ジャッジ時間 | 17,105 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
from collections import *from itertools import *from heapq import *from functools import *import sys,mathsys.setrecursionlimit(10**6)input = sys.stdin.readlineN = int(input())e = [[] for _ in range(N)]for _ in range(N-1):u,v = map(int,input().split())u -= 1v -= 1e[u].append(v)e[v].append(u)mod = 998244353parents = [-1]*Nsub = [-1]*Ndef dfs(x):tmp = 1if len(e[x]) == 1 and x != 0:sub[x] = 1return 1for ix in e[x]:if ix == parents[x]:continueparents[ix] = xif sub[ix] != -1:tmp += sub[ix]else:tmp += dfs(ix)sub[x] = tmpreturn tmpdfs(0)ans = 0N_inv = pow(N-1,mod-2,mod)M_inv = pow(N*(N-1)//2%mod,mod-2,mod)for i in range(1,N):ans += ((N-sub[i])*(N-sub[i]-1)//2%mod+(sub[i]*(sub[i]-1)//2)%mod)*M_inv%modans %= modans *= N_invans %= modprint(ans)