結果
問題 | No.1718 Random Squirrel |
ユーザー | tassei903 |
提出日時 | 2021-10-22 23:28:03 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 402 ms / 2,000 ms |
コード長 | 2,126 bytes |
コンパイル時間 | 1,237 ms |
コンパイル使用メモリ | 81,748 KB |
実行使用メモリ | 117,808 KB |
最終ジャッジ日時 | 2023-10-24 15:36:51 |
合計ジャッジ時間 | 8,693 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge13 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 38 ms
55,560 KB |
testcase_01 | AC | 38 ms
55,560 KB |
testcase_02 | AC | 38 ms
55,560 KB |
testcase_03 | AC | 39 ms
55,560 KB |
testcase_04 | AC | 39 ms
55,560 KB |
testcase_05 | AC | 40 ms
55,560 KB |
testcase_06 | AC | 39 ms
55,560 KB |
testcase_07 | AC | 39 ms
55,560 KB |
testcase_08 | AC | 39 ms
55,560 KB |
testcase_09 | AC | 39 ms
55,560 KB |
testcase_10 | AC | 39 ms
55,560 KB |
testcase_11 | AC | 264 ms
97,636 KB |
testcase_12 | AC | 171 ms
81,504 KB |
testcase_13 | AC | 249 ms
91,308 KB |
testcase_14 | AC | 238 ms
97,372 KB |
testcase_15 | AC | 279 ms
99,740 KB |
testcase_16 | AC | 233 ms
85,484 KB |
testcase_17 | AC | 243 ms
96,932 KB |
testcase_18 | AC | 336 ms
103,640 KB |
testcase_19 | AC | 300 ms
99,828 KB |
testcase_20 | AC | 165 ms
85,136 KB |
testcase_21 | AC | 313 ms
107,724 KB |
testcase_22 | AC | 382 ms
117,312 KB |
testcase_23 | AC | 387 ms
108,252 KB |
testcase_24 | AC | 309 ms
107,724 KB |
testcase_25 | AC | 402 ms
117,808 KB |
testcase_26 | AC | 192 ms
109,528 KB |
testcase_27 | AC | 204 ms
108,784 KB |
testcase_28 | AC | 177 ms
109,532 KB |
testcase_29 | AC | 189 ms
106,888 KB |
testcase_30 | AC | 193 ms
107,172 KB |
testcase_31 | AC | 171 ms
106,032 KB |
testcase_32 | AC | 183 ms
106,872 KB |
ソースコード
import sys input = lambda :sys.stdin.readline()[:-1] ni = lambda :int(input()) na = lambda :list(map(int,input().split())) sys.setrecursionlimit(10**7) yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES") no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO") ####################################################################### # 全方位木DP import sys input = sys.stdin.readline from collections import deque N,K = na() X = [[] for i in range(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) D = na() MO = 10**7 DD = [0 for i in range(N)] for i in D: DD[i-1] = MO P = [-1] * N Q = deque([0]) R = [] while Q: i = deque.popleft(Q) R.append(i) for a in X[i]: if a != P[i]: P[a] = i X[a].remove(i) deque.append(Q, a) ##### Settings unit = (0,0) merge = lambda a, b: (max(a[0],b[0],a[0]+b[0]),max(a[1],b[1],a[0]+b[1],a[1]+b[0])) adj_bu = lambda a, i: (a[0]-2+DD[i],a[1]-1+DD[i]) adj_td = lambda a, i, p: (a[0]-2+DD[p],a[1]-1+DD[p]) adj_fin = lambda a, i: (a[0]+DD[i],a[1]+DD[i]) ##### ME = [unit] * N # 始めはres_bu[i](pを除いたもの)、のちにres[i] XX = [(1,0)] * N TD = [unit] * N for i in R[1:][::-1]: XX[i] = adj_bu(ME[i], i) p = P[i] ME[p] = merge(ME[p], XX[i]) XX[R[0]] = adj_fin(ME[R[0]], R[0]) #print("ME =", ME) # Merge before adj #print("XX =", XX) # Bottom-up after adj for p in R: # 左からとres_td[p] ac = TD[p] for i in X[p]: # 左累積res_bu[c]とres_td[p]を入れとく TD[i] = ac # 左累積にres_bu[i]を追加する ac = merge(ac, XX[i]) # 右から ac = unit for i in X[p][::-1]: # TD[i]を確定する TD[i] = adj_td(merge(TD[i], ac), i, p) # 右累積に追加 ac = merge(ac, XX[i]) # res[i]を確定する ME[i](子側)TD[i](親側) XX[i] = adj_fin(merge(ME[i], TD[i]), i) #print("TD =", TD) # Top-down after adj #print("XX =", XX) # Final Result print(*[MO*K-i[1] for i in XX], sep = "\n")