結果

問題 No.876 Range Compress Query
ユーザー nayuta999999
提出日時 2020-03-05 19:42:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 389 ms / 2,000 ms
コード長 1,607 bytes
コンパイル時間 150 ms
コンパイル使用メモリ 82,064 KB
実行使用メモリ 94,412 KB
最終ジャッジ日時 2024-10-14 02:05:16
合計ジャッジ時間 4,671 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

#####segfunc######
def segfunc(x,y):
    return x + y

def init(init_val):
    #set_val
    for i in range(len(init_val)):
        seg[i+num-1]=init_val[i]   
    #built
    for i in range(num-2,-1,-1) :
        seg[i]=segfunc(seg[2*i+1],seg[2*i+2]) 
    
def update(k,x):
    k += num-1
    seg[k] = x
    while k:
        k = (k-1)//2
        seg[k] = segfunc(seg[k*2+1],seg[k*2+2])
    
def query(p,q):
    if q<=p:
        return ide_ele
    p += num-1
    q += num-2
    res=ide_ele
    while q-p>1:
        if p&1 == 0:
            res = segfunc(res,seg[p])
        if q&1 == 1:
            res = segfunc(res,seg[q])
            q -= 1
        p = p//2
        q = (q-1)//2
    if p == q:
        res = segfunc(res,seg[p])
    else:
        res = segfunc(segfunc(res,seg[p]),seg[q])
    return res
n, q = map(int, input().split())
#####単位元######
ide_ele = 0
num =2**(n-1).bit_length()
seg=[ide_ele]*(2*num - 1)
#阪堺区間に注意
a = list(map(int, input().split()))
b = [a[i+1] - a[i] for i in range(n-1)]
c = [1 if b[i] != 0 else 0 for i in range(len(b))]
init(c)
for i in range(q):
    query_input = list(map(int, input().split()))
    if query_input[0] == 1:
        l, r, x = query_input[1:]
        l,r = l-1, r-1
        if l > 0:
            b[l-1] += x
            if b[l-1] != 0:
                update(l-1,1)
            else:
                update(l-1,0)
        if r < n-1:
            b[r] -= x
            if b[r] != 0:
                update(r,1)
            else:
                update(r,0)
        
    else:
        l, r = query_input[1:]
        print(query(l-1,r-1)+1)
0