結果
問題 |
No.1098 LCAs
|
ユーザー |
![]() |
提出日時 | 2020-07-24 02:11:56 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 1,588 ms / 2,000 ms |
コード長 | 1,240 bytes |
コンパイル時間 | 273 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 113,956 KB |
最終ジャッジ日時 | 2024-06-24 02:09:50 |
合計ジャッジ時間 | 19,249 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import sys; input = sys.stdin.buffer.readline sys.setrecursionlimit(10**7) from collections import defaultdict con = 10 ** 9 + 7; INF = float("inf") def getlist(): return list(map(int, input().split())) class Graph(object): def __init__(self): self.graph = defaultdict(list) def __len__(self): return len(self.graph) def add_edge(self, a, b): self.graph[a].append(b) def DFS(G, W, Wlist, visit, node): for i in G.graph[node]: if visit[i] != 1: visit[i] = 1 DFS(G, W, Wlist, visit, i) W[node] += W[i] Wlist[node].append(W[i]) #処理内容 def main(): N = int(input()) G = Graph() for i in range(N - 1): a, b = getlist() a -= 1; b -= 1 G.add_edge(a, b) G.add_edge(b, a) W = [1] * N Wlist =[[] for i in range(N)] # print(len(Wlist)) Wans = [0] * N visit = [0] * N visit[0] = 1 DFS(G, W, Wlist, visit, 0) # print(Wlist) for i in range(N): if Wlist[i] == []: Wans[i] = 1 else: Wans[i] = (W[i] - 1) * (W[i] - 2) + 2 * (W[i] - 1) + 1 for j in Wlist[i]: Wans[i] -= j * (j - 1) # else: # Wans[i] = 2 * ((W[i] - 1) ** 2) + 2 * (W[i] - 1) + 1 # for j in Wlist[i]: # Wans[i] -= 2 * (j ** 2) for i in range(N): print(Wans[i]) if __name__ == '__main__': main()