結果
| 問題 | No.1103 Directed Length Sum |
| ユーザー |
👑  rin204
|
| 提出日時 | 2020-08-13 07:50:05 |
| 言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
| 結果 |
TLE
|
| 実行時間 | - |
| コード長 | 538 bytes |
| コンパイル時間 | 421 ms |
| コンパイル使用メモリ | 12,672 KB |
| 実行使用メモリ | 158,752 KB |
| 最終ジャッジ日時 | 2024-10-09 18:59:17 |
| 合計ジャッジ時間 | 6,051 ms |
|
ジャッジサーバーID （参考情報） |
judge1 / judge2 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 31 ms
17,696 KB |
| testcase_01 | AC | 32 ms
10,752 KB |
| testcase_02 | TLE | - |
| testcase_03 | -- | - |
| testcase_04 | -- | - |
| testcase_05 | -- | - |
| testcase_06 | -- | - |
| testcase_07 | -- | - |
| testcase_08 | -- | - |
| testcase_09 | -- | - |
| testcase_10 | -- | - |
| testcase_11 | -- | - |
| testcase_12 | -- | - |
| testcase_13 | -- | - |
| testcase_14 | -- | - |
| testcase_15 | -- | - |
| testcase_16 | -- | - |
| testcase_17 | -- | - |
| testcase_18 | -- | - |
| testcase_19 | -- | - |
| testcase_20 | -- | - |
| testcase_21 | -- | - |
| testcase_22 | -- | - |
| testcase_23 | -- | - |
ソースコード
from collections import deque
MOD = 10 ** 9 + 7
n = int(input())
root_cand = [True for _ in range(n)]
edges = [[] for _ in range(n)]
for _ in range(n - 1):
a, b = map(int, input().split())
edges[a - 1].append(b - 1)
root_cand[b - 1] = False
root = root_cand.index(True)
ans = 0
queue = deque()
queue.append(root)
queue.append(0)
while queue:
pos = queue.popleft()
d = queue.popleft()
ans += d * (d + 1) // 2
for n_pos in edges[pos]:
queue.append(n_pos)
queue.append(d + 1)
print(ans)