結果

問題 No.877 Range ReLU Query
コンテスト
ユーザー chineristAC
提出日時 2020-06-18 04:01:04
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 981 ms / 2,000 ms
コード長 1,095 bytes
記録
コンパイル時間 391 ms
コンパイル使用メモリ 82,944 KB
実行使用メモリ 108,144 KB
最終ジャッジ日時 2024-11-08 10:28:43
合計ジャッジ時間 11,207 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

class BIT():
    def __init__(self,n):
        self.BIT=[0]*(n+1)
        self.num=n

    def Query(self,idx):
        res_sum = 0
        while idx > 0:
            res_sum += self.BIT[idx]
            idx -= idx&(-idx)
        return res_sum

    #Ai += x O(logN)
    def update(self,idx,x):
        while idx <= self.num:
            self.BIT[idx] += x
            idx += idx&(-idx)
        return

import sys

input=sys.stdin.buffer.readline

N,Q=map(int,input().split())
A=list(map(int,input().split()))
query=[(A[i],-1,i) for i in range(N)]
ans=[0]*Q
for i in range(Q):
    id,l,r,x=map(int,input().split())
    query.append((x,l,r,i))

query.sort()
bitsum=BIT(N)
bitzero=BIT(N)
for i in range(N):
    bitsum.update(i+1,A[i])
    bitzero.update(i+1,1)
for i in range(N+Q):
    if len(query[i])==3:
        val,gomi,id=query[i]
        bitsum.update(id+1,-A[id])
        bitzero.update(id+1,-1)
    else:
        x,l,r,id=query[i]
        S=bitsum.Query(r)-bitsum.Query(l-1)
        minus=x*(bitzero.Query(r)-bitzero.Query(l-1))
        ans[id]=S-minus

for i in range(Q):
    print(ans[i])
0