結果

問題 No.2065 Sum of Min
ユーザー とりゐ
提出日時 2022-04-30 21:22:45
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 884 ms / 2,000 ms
コード長 1,578 bytes
コンパイル時間 449 ms
コンパイル使用メモリ 82,168 KB
実行使用メモリ 157,912 KB
最終ジャッジ日時 2024-11-06 19:17:53
合計ジャッジ時間 15,327 ms
ジャッジサーバーID
(参考情報)
judge2 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

from sys import stdin
input=lambda :stdin.readline()[:-1]
class segtree():
def __init__(self,init,func,ide):
self.n=len(init)
self.func=func
self.ide=ide
self.size=1<<(self.n-1).bit_length()
self.tree=[self.ide for i in range(2*self.size)]
for i in range(self.n):
self.tree[self.size+i]=init[i]
for i in range(self.size-1,0,-1):
self.tree[i]=self.func(self.tree[2*i], self.tree[2*i|1])
def update(self,k,x):
k+=self.size
self.tree[k]=x
k>>=1
while k:
self.tree[k]=self.func(self.tree[2*k],self.tree[k*2|1])
k>>=1
def get(self,i):
return self.tree[i+self.size]
def query(self,l,r):
l+=self.size
r+=self.size
l_res=self.ide
r_res=self.ide
while l<r:
if l&1:
l_res=self.func(l_res,self.tree[l])
l+=1
if r&1:
r-=1
r_res=self.func(self.tree[r],r_res)
l>>=1
r>>=1
return self.func(l_res,r_res)
def debug(self,s=10):
print([self.get(i) for i in range(min(self.n,s))])
from collections import defaultdict
n,q=map(int,input().split())
a=list(map(int,input().split()))
d=defaultdict(list)
s=set(a)
for i in range(n):
d[a[i]].append((i,-1,-1))
for i in range(q):
l,r,x=map(int,input().split())
d[x].append((l-1,r,i))
s.add(x)
def op(x,y):return x+y
seg1=segtree([0]*n,op,0)
seg2=segtree([1]*n,op,0)
ans=[0]*q
for i in sorted(list(s)):
for x,y,id in d[i]:
if id==-1:
seg1.update(x,i)
seg2.update(x,0)
else:
ans[id]=seg1.query(x,y)+seg2.query(x,y)*i
print(*ans,sep='\n')
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0