結果
問題 | No.2101 [Cherry Alpha N] ずっとこの数列だったらいいのに |
ユーザー |
![]() |
提出日時 | 2022-10-15 05:05:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 3,987 ms / 6,000 ms |
コード長 | 1,753 bytes |
コンパイル時間 | 593 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 363,564 KB |
最終ジャッジ日時 | 2024-11-19 07:51:29 |
合計ジャッジ時間 | 116,470 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 42 |
ソースコード
import sysimport io, osinput = io.BytesIO(os.read(0,os.fstat(0).st_size)).readlinefrom operator import itemgetterN=int(input())AT=[list(map(int,input().split())) for i in range(N)]Q=int(input())QU=[list(map(int,input().split()))+[i] for i in range(Q)]# Segment tree(1-indexed,再帰を使わないもの)def seg_function(x,y): # Segment treeで扱うfunctionreturn [x[0]+y[0],x[1]+y[1],x[2]+y[2]]seg_el=1<<(N.bit_length()) # Segment treeの台の要素数SEG=[[0,0,0] for i in range(2*seg_el)] # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化for i in range(N): # Aを対応する箇所へupdateSEG[i+seg_el]=[AT[i][0],0,0]for i in range(seg_el-1,0,-1): # 親の部分もupdateSEG[i]=seg_function(SEG[i*2],SEG[i*2+1])def update(n,x,seg_el): # A[n]をxへ更新i=n+seg_elSEG[i]=xi>>=1 # 子ノードへwhile i!=0:SEG[i]=seg_function(SEG[i*2],SEG[i*2+1])i>>=1def getvalues(l,r): # 区間[l,r)に関するseg_functionを調べるL=l+seg_elR=r+seg_elANS=[0,0,0]while L<R:if L & 1:ANS=seg_function(ANS , SEG[L])L+=1if R & 1:R-=1ANS=seg_function(ANS , SEG[R])L>>=1R>>=1return ANSfor i in range(N):a,t=AT[i]if a==0:continueQU.append([t,a,i])QU.append([t+a,i])QU.reverse()QU.sort(key=itemgetter(0))ANS=[-1]*Qfor q in QU:if len(q)==4:d,l,r,ind=qx,y,z=getvalues(l-1,r)ANS[ind]=x-d*z+yelif len(q)==3:t,a,i=qupdate(i,[a,t-1,1],seg_el)else:t,i=qupdate(i,[0,0,0],seg_el)print("\n".join(map(str,ANS)))