結果

問題 No.900 aδδitivee
コンテスト
ユーザー Kiri8128
提出日時 2019-10-05 16:08:37
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
TLE  
実行時間 -
コード長 2,214 bytes
コンパイル時間 188 ms
コンパイル使用メモリ 13,056 KB
実行使用メモリ 66,720 KB
最終ジャッジ日時 2024-10-06 03:00:41
合計ジャッジ時間 4,755 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 5 TLE * 1 -- * 21
権限があれば一括ダウンロードができます

ソースコード

diff # 

NN = 17
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))
0