結果
問題 | No.1103 Directed Length Sum |
ユーザー | Kiri8128 |
提出日時 | 2020-07-03 22:15:16 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,910 ms / 3,000 ms |
コード長 | 646 bytes |
コンパイル時間 | 1,108 ms |
コンパイル使用メモリ | 81,768 KB |
実行使用メモリ | 302,116 KB |
最終ジャッジ日時 | 2023-10-17 04:14:42 |
合計ジャッジ時間 | 24,115 ms |
ジャッジサーバーID (参考情報) |
judge14 / judge15 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 40 ms
55,580 KB |
testcase_01 | AC | 39 ms
55,580 KB |
testcase_02 | AC | 811 ms
259,272 KB |
testcase_03 | AC | 929 ms
302,116 KB |
testcase_04 | AC | 1,497 ms
204,828 KB |
testcase_05 | AC | 2,910 ms
262,888 KB |
testcase_06 | AC | 906 ms
161,912 KB |
testcase_07 | AC | 223 ms
96,056 KB |
testcase_08 | AC | 327 ms
107,216 KB |
testcase_09 | AC | 163 ms
86,812 KB |
testcase_10 | AC | 453 ms
118,576 KB |
testcase_11 | AC | 1,621 ms
209,536 KB |
testcase_12 | AC | 909 ms
150,932 KB |
testcase_13 | AC | 440 ms
120,924 KB |
testcase_14 | AC | 139 ms
84,996 KB |
testcase_15 | AC | 697 ms
134,740 KB |
testcase_16 | AC | 1,889 ms
216,868 KB |
testcase_17 | AC | 2,008 ms
221,144 KB |
testcase_18 | AC | 447 ms
118,564 KB |
testcase_19 | AC | 1,698 ms
204,756 KB |
testcase_20 | AC | 178 ms
91,384 KB |
testcase_21 | AC | 292 ms
104,448 KB |
testcase_22 | AC | 1,385 ms
184,648 KB |
testcase_23 | AC | 767 ms
149,440 KB |
ソースコード
from collections import deque N = int(input()) X = [[] for i in range(N)] Y = [0] * N for i in range(N-1): x, y = map(int, input().split()) X[x-1].append(y-1) X[y-1].append(x-1) Y[y-1] += 1 for i in range(N): if Y[i] == 0: s = i break P = [-1] * N Q = deque([s]) D = [0] * N R = [] while Q: i = deque.popleft(Q) R.append(i) for a in X[i]: if a != P[i]: P[a] = i D[a] = D[i] + 1 X[a].remove(i) deque.append(Q, a) mod = 10 ** 9 + 7 C = [1] * N ans = 0 for i in R[1:][::-1]: C[P[i]] += C[i] ans = (ans + D[i] * C[i]) % mod print(ans)