結果
問題 |
No.1098 LCAs
|
ユーザー |
![]() |
提出日時 | 2020-05-25 03:07:06 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,785 ms / 2,000 ms |
コード長 | 689 bytes |
コンパイル時間 | 101 ms |
コンパイル使用メモリ | 12,672 KB |
実行使用メモリ | 114,276 KB |
最終ジャッジ日時 | 2024-11-17 21:52:34 |
合計ジャッジ時間 | 24,244 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import sys read = sys.stdin.buffer.read readline = sys.stdin.buffer.readline readlines = sys.stdin.buffer.readlines N = int(readline()) UV = [map(int, readline().split()) for _ in range(N-1)] graph = [[] for _ in range(N)] for u, v in UV: graph[u-1].append(v-1) graph[v-1].append(u-1) stack = [0] dfs_ord = [] par = [-1 for _ in range(N)] while stack: v = stack.pop() dfs_ord.append(v) for u in graph[v]: if u != par[v]: par[u] = v stack.append(u) sz = [1 for _ in range(N)] for v in dfs_ord[1:][::-1]: sz[par[v]] += sz[v] for v in range(N): sz[v] *= sz[v] res = sz.copy() for v in range(1, N): res[par[v]] -= sz[v] for r in res: print(r)