結果
問題 | No.2892 Lime and Karin |
ユーザー | adapchi |
提出日時 | 2024-09-13 23:04:05 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,421 bytes |
コンパイル時間 | 254 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 198,984 KB |
最終ジャッジ日時 | 2024-09-13 23:04:17 |
合計ジャッジ時間 | 11,472 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 44 ms
52,952 KB |
testcase_01 | AC | 38 ms
53,356 KB |
testcase_02 | AC | 39 ms
53,292 KB |
testcase_03 | AC | 39 ms
54,052 KB |
testcase_04 | AC | 41 ms
53,904 KB |
testcase_05 | AC | 48 ms
60,988 KB |
testcase_06 | AC | 48 ms
61,028 KB |
testcase_07 | AC | 40 ms
54,672 KB |
testcase_08 | AC | 65 ms
67,936 KB |
testcase_09 | AC | 57 ms
66,120 KB |
testcase_10 | AC | 53 ms
63,360 KB |
testcase_11 | AC | 40 ms
54,156 KB |
testcase_12 | AC | 42 ms
54,392 KB |
testcase_13 | AC | 47 ms
59,616 KB |
testcase_14 | TLE | - |
testcase_15 | -- | - |
testcase_16 | -- | - |
testcase_17 | -- | - |
testcase_18 | -- | - |
testcase_19 | -- | - |
testcase_20 | -- | - |
testcase_21 | -- | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
testcase_29 | -- | - |
testcase_30 | -- | - |
testcase_31 | -- | - |
testcase_32 | -- | - |
testcase_33 | -- | - |
testcase_34 | -- | - |
testcase_35 | -- | - |
testcase_36 | -- | - |
testcase_37 | -- | - |
testcase_38 | -- | - |
testcase_39 | -- | - |
testcase_40 | -- | - |
testcase_41 | -- | - |
testcase_42 | -- | - |
testcase_43 | -- | - |
testcase_44 | -- | - |
testcase_45 | -- | - |
testcase_46 | -- | - |
testcase_47 | -- | - |
testcase_48 | -- | - |
testcase_49 | -- | - |
testcase_50 | -- | - |
testcase_51 | -- | - |
testcase_52 | -- | - |
testcase_53 | -- | - |
testcase_54 | -- | - |
ソースコード
from bisect import bisect_left N = int(input()) G = [[] for _ in range(N)] for _ in range(N - 1): u, v = [int(x) - 1 for x in input().split()] G[u].append(v) G[v].append(u) S = [1 if x == '1' else -1 for x in input()] used = [False] * N size = [0] * N def centroid(x): stk = [(x, None, 0)] while stk: u, p, t = stk.pop() if t == 0: stk.append((u, p, 1)) for v in G[u]: if not used[v] and v != p: stk.append((v, u, 0)) else: size[u] = 1 + sum(size[v] for v in G[u] if not used[v] and v != p) p = None while True: y = next((y for y in G[x] if not used[y] and y != p and size[y] * 2 > size[x]), None) if y is None: break x, p = y, x return x def totals(x, p): stk = [(x, p, S[x])] res = [] while stk: u, p, d = stk.pop() res.append(d) for v in G[u]: if not used[v] and v != p: stk.append((v, u, d + S[v])) return res stk = [0] ans = 0 while stk: u = centroid(stk.pop()) ds = [] val = 0 if S[u] == 1: ans += 1 for v in G[u]: if used[v]: continue stk.append(v) d = totals(v, u) ans += sum(x + S[u] > 0 for x in d) d.sort() # d[i] + d[j] + S[u] > 0 val -= sum(len(d) - bisect_left(d, -x - S[u] + 1) for x in d) ds.extend(d) ds.sort() val += sum(len(ds) - bisect_left(ds, -x - S[u] + 1) for x in ds) ans += val // 2 used[u] = True print(ans)