結果
問題 | No.1103 Directed Length Sum |
ユーザー | brthyyjp |
提出日時 | 2020-09-09 02:52:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,113 ms / 3,000 ms |
コード長 | 946 bytes |
コンパイル時間 | 628 ms |
コンパイル使用メモリ | 82,340 KB |
実行使用メモリ | 382,432 KB |
最終ジャッジ日時 | 2024-05-07 23:12:39 |
合計ジャッジ時間 | 16,253 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 34 ms
52,096 KB |
testcase_01 | AC | 33 ms
52,352 KB |
testcase_02 | AC | 759 ms
379,648 KB |
testcase_03 | AC | 656 ms
382,432 KB |
testcase_04 | AC | 930 ms
220,888 KB |
testcase_05 | AC | 2,113 ms
306,216 KB |
testcase_06 | AC | 599 ms
162,856 KB |
testcase_07 | AC | 138 ms
98,688 KB |
testcase_08 | AC | 183 ms
107,008 KB |
testcase_09 | AC | 98 ms
87,168 KB |
testcase_10 | AC | 265 ms
123,904 KB |
testcase_11 | AC | 969 ms
229,300 KB |
testcase_12 | AC | 566 ms
170,368 KB |
testcase_13 | AC | 331 ms
122,752 KB |
testcase_14 | AC | 123 ms
85,380 KB |
testcase_15 | AC | 451 ms
151,588 KB |
testcase_16 | AC | 1,121 ms
246,856 KB |
testcase_17 | AC | 1,170 ms
252,096 KB |
testcase_18 | AC | 284 ms
120,704 KB |
testcase_19 | AC | 1,077 ms
232,960 KB |
testcase_20 | AC | 156 ms
89,216 KB |
testcase_21 | AC | 165 ms
104,832 KB |
testcase_22 | AC | 805 ms
212,096 KB |
testcase_23 | AC | 453 ms
158,208 KB |
ソースコード
import sys import io, os input = sys.stdin.buffer.readline #input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline mod = 10**9+7 def main(): n = int(input()) g = [[] for _ in range(n)] ind = [[] for _ in range(n)] for i in range(n-1): a, b = map(int, input().split()) a, b = a-1, b-1 g[a].append(b) ind[b].append(a) root = -1 for i in range(n): if len(ind[i]) == 0: root = i break s = [] s.append(root) order = [] par = [-1]*n depth = [0]*n while s: v = s.pop() order.append(v) for u in g[v]: if u == par[v]: continue par[u] = v s.append(u) depth[u] = depth[v]+1 #print(depth) ans = 0 for i in range(n): d = depth[i] ans += d*(d+1)//2 ans %= mod print(ans) if __name__ == '__main__': main()