結果
問題 | No.1103 Directed Length Sum |
ユーザー |
![]() |
提出日時 | 2020-07-07 15:36:47 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 800 bytes |
コンパイル時間 | 217 ms |
コンパイル使用メモリ | 82,192 KB |
実行使用メモリ | 396,312 KB |
最終ジャッジ日時 | 2024-10-01 08:17:43 |
合計ジャッジ時間 | 25,006 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 21 WA * 1 |
ソースコード
MOD = 1000000007N = int(input())edge = [tuple(map(int, input().split())) for _ in range(N - 1)]tree = [[] for _ in range(N)]par = [None for _ in range(N)]dep = [None for _ in range(N)]cum = [None for _ in range(N)]for e in edge:tree[e[0] - 1].append(e[1] - 1)par[e[1] - 1] = e[0] - 1for i in range(N):if par[i] is None:root = istack = [root]dep[root] = 0while stack:node = stack.pop()for adj in tree[node]:if dep[adj] is not None:continuedep[adj] = dep[node] + 1stack.append(adj)stack = [root]cum[root] = 0while stack:node = stack.pop()for adj in tree[node]:if cum[adj] is not None:continuecum[adj] = cum[node] + dep[adj]stack.append(adj)print(sum(cum))