結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2020-09-13 05:38:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 499 ms / 2,000 ms |
コード長 | 755 bytes |
コンパイル時間 | 436 ms |
コンパイル使用メモリ | 82,596 KB |
実行使用メモリ | 127,628 KB |
最終ジャッジ日時 | 2025-01-03 00:47:04 |
合計ジャッジ時間 | 8,807 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import sysinput=sys.stdin.readlinedef dfs(start):ans = [0]*Nstack = [start]parent = [N]*Npart_num = [1]*Nparent[start] = -1while stack:v = stack[-1]marker = 0cnt = 1for u in edge[v]:if u == parent[v]:continueif parent[u]==N: #前半marker = 1parent[u] = vstack.append(u)else: #後半ans[v] -= part_num[u]**2part_num[v] += part_num[u]if marker==0: #後半だったらstack.pop()ans[v] += part_num[v]**2return ansN = int(input())edge = [[] for _ in range(N)]visited = [False]*Nfor _ in range(N-1):a,b = map(int, input().split())edge[a-1].append(b-1)edge[b-1].append(a-1)ans = dfs(0)print(*ans,sep='\n')