結果
問題 |
No.1103 Directed Length Sum
|
ユーザー |
|
提出日時 | 2020-07-03 22:12:39 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,216 ms / 3,000 ms |
コード長 | 563 bytes |
コンパイル時間 | 344 ms |
コンパイル使用メモリ | 81,792 KB |
実行使用メモリ | 462,840 KB |
最終ジャッジ日時 | 2024-09-17 02:42:48 |
合計ジャッジ時間 | 20,817 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 22 |
ソースコード
from collections import defaultdict MOD = 10**9 + 7 inpl = lambda: list(map(int,input().split())) N = int(input()) children = [set() for _ in range(N)] parent = [None] * N for _ in range(N-1): A, B = inpl() A, B = A-1, B-1 children[A].add(B) parent[B] = A layer = defaultdict(int) forward, back = 0, 1 root = parent.index(None) pool = [(root,0)] while pool: node, d = pool.pop() layer[d] += 1 for c in children[node]: pool.append((c, d+1)) ans = 0 for d, n in layer.items(): ans += n*d*(d+1)//2 ans %= MOD print(ans)