結果
問題 | 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 pypyjitpypyjit.set_param('max_unroll_recursion=-1')from collections import *from itertools import *from functools import *from heapq import *import sys,mathsys.setrecursionlimit(10**6)input = sys.stdin.readlineN = int(input())e = [[] for _ in range(N)]for _ in range(N-1):a,b = map(int,input().split())a -= 1b -= 1e[a].append(b)e[b].append(a)X = [[] for _ in range(N)]vis = [False]*Nvis[0]=Truev = deque()v.append(0)parents = [-1]*Nsub = [-1]*Ndef dfs(x):tmp = 1if len(e[x]) == 1 and x != 0:sub[x] = 1return 1for ix in e[x]:if ix == parents[x]:continueparents[ix] = xif sub[ix] != -1:tmp += sub[ix]else:tmp += dfs(ix)sub[x] = tmpreturn tmpdfs(0)while v:x = v.popleft()for ix in e[x]:if vis[ix]:continueX[x].append(sub[ix])vis[ix]=Truev.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 + 1print(ans)# print(X)