結果
問題 | No.1507 Road Blocked |
ユーザー | nasubi24 |
提出日時 | 2021-05-14 22:42:42 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 375 ms / 2,000 ms |
コード長 | 547 bytes |
コンパイル時間 | 184 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 181,168 KB |
最終ジャッジ日時 | 2024-10-02 02:49:17 |
合計ジャッジ時間 | 12,828 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 30 |
ソースコード
import sys sys.setrecursionlimit(10**6) n=int(input()) mod=998244353 q=pow((n-1)*(n-1)*n//2,mod-2,mod) path=[[] for _ in range(n)] p2=[] for i in range(n-1): a,b=map(int,input().split()) path[a-1].append(b-1) path[b-1].append(a-1) p2.append([a-1,b-1]) s=[1]*n seen=[False]*n def dfs(t): global seen,s for i in path[t]: if not seen[i]: seen[i]=True s[t]+=dfs(i) return s[t] seen[0]=True s[0]=dfs(0) p=0 m=(n-1)*n//2 for i in p2: t=min(s[i[0]],s[i[1]]) p+=m-t*(n-t) print(p*q%mod)