結果
問題 | No.1103 Directed Length Sum |
ユーザー | marroncastle |
提出日時 | 2020-12-31 18:51:53 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,391 ms / 3,000 ms |
コード長 | 1,190 bytes |
コンパイル時間 | 1,463 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 218,972 KB |
最終ジャッジ日時 | 2024-10-09 18:04:29 |
合計ジャッジ時間 | 16,021 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 40 ms
53,812 KB |
testcase_01 | AC | 43 ms
53,940 KB |
testcase_02 | AC | 568 ms
207,420 KB |
testcase_03 | AC | 478 ms
218,972 KB |
testcase_04 | AC | 750 ms
133,268 KB |
testcase_05 | AC | 1,391 ms
172,252 KB |
testcase_06 | AC | 495 ms
113,664 KB |
testcase_07 | AC | 146 ms
85,248 KB |
testcase_08 | AC | 191 ms
89,856 KB |
testcase_09 | AC | 115 ms
82,048 KB |
testcase_10 | AC | 250 ms
95,232 KB |
testcase_11 | AC | 830 ms
138,624 KB |
testcase_12 | AC | 486 ms
113,920 KB |
testcase_13 | AC | 267 ms
95,616 KB |
testcase_14 | AC | 104 ms
80,128 KB |
testcase_15 | AC | 384 ms
105,856 KB |
testcase_16 | AC | 941 ms
145,664 KB |
testcase_17 | AC | 992 ms
148,864 KB |
testcase_18 | AC | 238 ms
94,848 KB |
testcase_19 | AC | 864 ms
140,032 KB |
testcase_20 | AC | 127 ms
83,200 KB |
testcase_21 | AC | 180 ms
88,448 KB |
testcase_22 | AC | 686 ms
128,896 KB |
testcase_23 | AC | 394 ms
107,904 KB |
ソースコード
import sys input = sys.stdin.readline from collections import deque MOD = 10**9+7 class Tree: def __init__(self, N): self.V = N self.edge = [[] for _ in range(N)] self.to = [False]*N def add_edges(self, ind=1, bi=True): for i in range(self.V-1): a,b = map(int, input().split()) a -= ind; b -= ind self.edge[a].append(b) self.to[b] = True if bi: self.edge[b].append(a) def add_edge(self, a, b, bi=True): self.edge[a].append(b) if bi: self.edge[b].append(a) def dfs(self): for v in range(self.V): if not self.to[v]: start = v break stack = deque([start]) self.parent = [N]*N self.parent[start] = -1 cnt = 0 self.depth = [-1]*self.V self.depth[start] = 0 #記録したい値の配列を定義 while stack: v = stack.pop() for u in self.edge[v]: if u==self.parent[v]: continue self.parent[u]=v stack.append(u) self.depth[u] = self.depth[v]+1 cnt += self.depth[u]*(self.depth[u]+1)//2 cnt %= MOD return cnt N = int(input()) G = Tree(N) G.add_edges(ind=1, bi=False) print(G.dfs())