結果
問題 | No.1103 Directed Length Sum |
ユーザー | terasa |
提出日時 | 2022-06-09 01:43:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 2,843 ms / 3,000 ms |
コード長 | 1,557 bytes |
コンパイル時間 | 250 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 296,192 KB |
最終ジャッジ日時 | 2024-09-21 05:22:09 |
合計ジャッジ時間 | 23,469 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 47 ms
55,168 KB |
testcase_01 | AC | 46 ms
54,784 KB |
testcase_02 | AC | 991 ms
296,192 KB |
testcase_03 | AC | 809 ms
273,280 KB |
testcase_04 | AC | 1,488 ms
193,268 KB |
testcase_05 | AC | 2,843 ms
273,272 KB |
testcase_06 | AC | 983 ms
151,072 KB |
testcase_07 | AC | 251 ms
92,688 KB |
testcase_08 | AC | 371 ms
103,232 KB |
testcase_09 | AC | 181 ms
87,544 KB |
testcase_10 | AC | 488 ms
113,180 KB |
testcase_11 | AC | 1,655 ms
203,392 KB |
testcase_12 | AC | 962 ms
151,936 KB |
testcase_13 | AC | 503 ms
114,176 KB |
testcase_14 | AC | 151 ms
83,712 KB |
testcase_15 | AC | 743 ms
135,680 KB |
testcase_16 | AC | 1,885 ms
217,472 KB |
testcase_17 | AC | 1,951 ms
221,684 KB |
testcase_18 | AC | 494 ms
113,152 KB |
testcase_19 | AC | 1,707 ms
204,288 KB |
testcase_20 | AC | 207 ms
88,924 KB |
testcase_21 | AC | 334 ms
99,840 KB |
testcase_22 | AC | 1,386 ms
182,016 KB |
testcase_23 | AC | 829 ms
137,600 KB |
ソースコード
import sys import pypyjit import itertools import heapq import math from collections import deque, defaultdict import bisect input = sys.stdin.readline sys.setrecursionlimit(10 ** 6) pypyjit.set_param('max_unroll_recursion=-1') def index_lt(a, x): 'return largest index s.t. A[i] < x or -1 if it does not exist' return bisect.bisect_left(a, x) - 1 def index_le(a, x): 'return largest index s.t. A[i] <= x or -1 if it does not exist' return bisect.bisect_right(a, x) - 1 def index_gt(a, x): 'return smallest index s.t. A[i] > x or len(a) if it does not exist' return bisect.bisect_right(a, x) def index_ge(a, x): 'return smallest index s.t. A[i] >= x or len(a) if it does not exist' return bisect.bisect_left(a, x) N = int(input()) mod = 10 ** 9 + 7 E = [[] for _ in range(N)] par = [-1] * N D = [0] * N for _ in range(N - 1): a, b = map(int, input().split()) a -= 1 b -= 1 E[a].append(b) D[b] += 1 par[b] = a for i in range(N): if D[i] == 0: s = i break dq = deque([s]) INF = 1 << 30 depth = [INF] * N depth[s] = 0 while dq: v = dq.popleft() for d in E[v]: if depth[d] > depth[v] + 1: depth[d] = depth[v] + 1 dq.append(d) V = [(i, depth[i]) for i in range(N)] V.sort(key=lambda x: x[1], reverse=True) cnt = [1] * N acc = [0] * N for i, _ in V: acc[i] += cnt[i] - 1 if i == s: break p = par[i] cnt[p] += cnt[i] acc[p] += acc[i] ans = 0 for i in range(N): ans += acc[i] ans %= mod print(ans)