結果
問題 | No.900 aδδitivee |
ユーザー | Kiri8128 |
提出日時 | 2019-10-05 16:07:45 |
言語 | Python3 (3.12.2 + numpy 1.26.4 + scipy 1.12.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 2,252 bytes |
コンパイル時間 | 158 ms |
コンパイル使用メモリ | 13,056 KB |
実行使用メモリ | 66,608 KB |
最終ジャッジ日時 | 2024-10-06 02:57:53 |
合計ジャッジ時間 | 22,782 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 27 ms
11,008 KB |
testcase_01 | AC | 29 ms
11,136 KB |
testcase_02 | RE | - |
testcase_03 | RE | - |
testcase_04 | RE | - |
testcase_05 | RE | - |
testcase_06 | RE | - |
testcase_07 | RE | - |
testcase_08 | RE | - |
testcase_09 | RE | - |
testcase_10 | RE | - |
testcase_11 | RE | - |
testcase_12 | RE | - |
testcase_13 | RE | - |
testcase_14 | RE | - |
testcase_15 | RE | - |
testcase_16 | RE | - |
testcase_17 | RE | - |
testcase_18 | RE | - |
testcase_19 | RE | - |
testcase_20 | RE | - |
testcase_21 | RE | - |
testcase_22 | RE | - |
testcase_23 | RE | - |
testcase_24 | RE | - |
testcase_25 | RE | - |
testcase_26 | RE | - |
testcase_27 | RE | - |
testcase_28 | RE | - |
ソースコード
from bisect import bisect_right as br NN = 5 BIT1=[0]*(2**NN+1) BIT2=[0]*(2**NN+1) def addrange(l0, r0, x=1): l, r = l0, r0 while l <= 2**NN: BIT1[l] -= x * l0 BIT2[l] -= x l += l & (-l) while r <= 2**NN: BIT1[r] += x * r0 BIT2[r] += x r += r & (-r) def rangesum(l, r): l0, r0, a, b, c = l, r, 0, 0, 0 while l != 0: a -= BIT1[l] b += BIT2[l] l -= l&(-l) while r != 0: a += BIT1[r] c += BIT2[r] r -= r&(-r) return a + b * l0 - c * r0 BIT3=[0]*(2**NN+1) BIT4=[0]*(2**NN+1) def addrange2(l0, r0, x=1): l, r = l0, r0 while l <= 2**NN: BIT3[l] -= x * l0 BIT4[l] -= x l += l & (-l) while r <= 2**NN: BIT3[r] += x * r0 BIT4[r] += x r += r & (-r) def rangesum2(l, r): l0, r0, a, b, c = l, r, 0, 0, 0 while l != 0: a -= BIT3[l] b += BIT4[l] l -= l&(-l) while r != 0: a += BIT3[r] c += BIT4[r] r -= r&(-r) return a + b * l0 - c * r0 N = int(input()) X = [[] for i in range(N)] for i in range(N-1): x, y, d = map(int, input().split()) X[x].append((y, d)) X[y].append((x, d)) P = [(-1, 0)] * N Q = [(-1, 0), (0, 0)] ET = [] ct = 0 dd = 0 CTA = [0] * N CTB = [0] * N DE = [0] * N D = [0] * N de = -1 while Q: i, d = Q.pop() if i < 0: CTB[~i] = ct de -= 1 dd -= d continue if i >= 0: ct += 1 if CTA[i] == 0: CTA[i] = ct dd += d de += 1 DE[i] = de D[i] = dd for a, d in X[i][::-1]: if a != P[i][0]: ET.append((i, a)) P[a] = (i, d) for k in range(len(X[a])): if X[a][k][0] == i: del X[a][k] break Q.append((~a, d)) Q.append((a, d)) for _ in range(int(input())): tmp = [int(a) for a in input().split()] if tmp[0] == 1: a, x = tmp[1:] addrange(CTA[a]+1, CTB[a]+1, x) addrange2(CTA[a]+1, CTB[a]+1, x * DE[a]) else: b = tmp[1] print(D[b] + rangesum(CTA[b], CTA[b]+1) * DE[b] - rangesum2(CTA[b], CTA[b]+1))