結果

問題 No.789 範囲の合計
ユーザー kiri8128
提出日時 2019-04-16 01:48:38
言語 Python3
(3.7.1 + numpy 1.14.5 + scipy 1.1.0)
結果
WA   .
実行時間 -
コード長 786 Byte
コンパイル時間 43 ms
使用メモリ 27,116 KB
最終ジャッジ日時 2019-06-05 20:54:18

テストケース

テストケース表示
入力 結果 実行時間
使用メモリ
input00.txt WA -
input01.txt AC 24 ms
7,384 KB
input02.txt TLE -
input03.txt -- -
input04.txt -- -
input05.txt -- -
input06.txt -- -
input07.txt -- -
input08.txt -- -
input09.txt -- -
input10.txt -- -
input11.txt -- -
input12.txt -- -
input13.txt -- -
input14.txt -- -
テストケース一括ダウンロード

ソースコード

diff #
from bisect import bisect_left as bl

N = 17
X = [0] * (2**(N+1)-1)

def add(j, x):
    i = 2**N + j - 1
    while i >= 0:
        X[i] += x
        i = (i-1) // 2

def rangeof(i):
    s = (len(bin(i+1))-3)
    l = ((i+1) - (1<<s)) * (1<<N-s)
    r = l + (1<<N-s)
    return (l, r)

def rangesum(a, b, i=0):
    l, r = rangeof(i)
    if l >= b or r <= a:
        return 0
    if a <= l and r <= b:
        return X[i]
    return rangesum(a, b, 2*i+1) + rangesum(a, b, 2*i+2)


Q = int(input())
s = 0
Y = []
Z = set([])
for _ in range(Q):
    t, a, b = map(int, input().split())
    Y.append((t, a, b))
    if t == 1:
        Z |= set([a, b])

Z = list(Z)

for t, a, b in Y:
    if t == 0:
        add(bl(Z, a), b)
    else:
        s += rangesum(bl(Z, a), bl(Z, b)+1)
    
    
print(s)
0