結果
問題 | No.1098 LCAs |
ユーザー |
|
提出日時 | 2023-09-21 13:52:24 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,177 ms / 2,000 ms |
コード長 | 1,155 bytes |
コンパイル時間 | 338 ms |
コンパイル使用メモリ | 82,116 KB |
実行使用メモリ | 287,232 KB |
最終ジャッジ日時 | 2024-07-07 07:36:33 |
合計ジャッジ時間 | 13,065 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import pypyjit pypyjit.set_param('max_unroll_recursion=-1') from collections import * from itertools import * from functools import * from heapq import * import sys,math sys.setrecursionlimit(10**6) input = sys.stdin.readline N = int(input()) e = [[] for _ in range(N)] for _ in range(N-1): a,b = map(int,input().split()) a -= 1 b -= 1 e[a].append(b) e[b].append(a) X = [[] for _ in range(N)] vis = [False]*N vis[0]=True v = deque() v.append(0) parents = [-1]*N sub = [-1]*N def dfs(x): tmp = 1 if len(e[x]) == 1 and x != 0: sub[x] = 1 return 1 for ix in e[x]: if ix == parents[x]: continue parents[ix] = x if sub[ix] != -1: tmp += sub[ix] else: tmp += dfs(ix) sub[x] = tmp return tmp dfs(0) while v: x = v.popleft() for ix in e[x]: if vis[ix]: continue X[x].append(sub[ix]) vis[ix]=True v.append(ix) for i in range(N): s2 = sum(j**2 for j in X[i]) s = sum(j for j in X[i]) ans = (s**2 - s2) + 2*s + 1 print(ans) # print(X)