結果
問題 |
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 sys import io, os input = io.BytesIO(os.read(0,os.fstat(0).st_size)).readline from operator import itemgetter N=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で扱うfunction return [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を対応する箇所へupdate SEG[i+seg_el]=[AT[i][0],0,0] for i in range(seg_el-1,0,-1): # 親の部分もupdate SEG[i]=seg_function(SEG[i*2],SEG[i*2+1]) def update(n,x,seg_el): # A[n]をxへ更新 i=n+seg_el SEG[i]=x i>>=1 # 子ノードへ while i!=0: SEG[i]=seg_function(SEG[i*2],SEG[i*2+1]) i>>=1 def getvalues(l,r): # 区間[l,r)に関するseg_functionを調べる L=l+seg_el R=r+seg_el ANS=[0,0,0] while L<R: if L & 1: ANS=seg_function(ANS , SEG[L]) L+=1 if R & 1: R-=1 ANS=seg_function(ANS , SEG[R]) L>>=1 R>>=1 return ANS for i in range(N): a,t=AT[i] if a==0: continue QU.append([t,a,i]) QU.append([t+a,i]) QU.reverse() QU.sort(key=itemgetter(0)) ANS=[-1]*Q for q in QU: if len(q)==4: d,l,r,ind=q x,y,z=getvalues(l-1,r) ANS[ind]=x-d*z+y elif len(q)==3: t,a,i=q update(i,[a,t-1,1],seg_el) else: t,i=q update(i,[0,0,0],seg_el) print("\n".join(map(str,ANS)))