結果
問題 | 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 |
ソースコード
from sys import stdininput=lambda :stdin.readline()[:-1]class segtree():def __init__(self,init,func,ide):self.n=len(init)self.func=funcself.ide=ideself.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.sizeself.tree[k]=xk>>=1while k:self.tree[k]=self.func(self.tree[2*k],self.tree[k*2|1])k>>=1def get(self,i):return self.tree[i+self.size]def query(self,l,r):l+=self.sizer+=self.sizel_res=self.ider_res=self.idewhile l<r:if l&1:l_res=self.func(l_res,self.tree[l])l+=1if r&1:r-=1r_res=self.func(self.tree[r],r_res)l>>=1r>>=1return 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 defaultdictn,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+yseg1=segtree([0]*n,op,0)seg2=segtree([1]*n,op,0)ans=[0]*qfor 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)*iprint(*ans,sep='\n')