結果

問題 No.1332 Range Nearest Query
ユーザー persimmon-persimmon
提出日時 2021-02-14 10:32:30
言語 PyPy3
(7.3.15)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 2,158 bytes
コンパイル時間 282 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 281,492 KB
最終ジャッジ日時 2024-07-21 16:15:47
合計ジャッジ時間 77,180 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 37 TLE * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=sys.stdin.readline

class SegmentTree():
  def __init__(self,size,f=lambda x,y:x+y,default=0):
    self.size=pow(2,(size-1).bit_length())
    self.f=f
    self.default=default
    self.data=[default]*(self.size*2)
  # list[i]をxに更新
  def update(self,i,x):
    i+=self.size
    self.data[i]=x
    while i:
      i>>=1
      self.data[i]=self.f(self.data[i*2],self.data[i*2+1])
  # はじめの構築をO(N)で行う
  def init_build(self,ary):
    n=len(ary)
    for i in range(n):
      self.data[i+self.size]=ary[i]
    for i in reversed(range(self.size)):
      self.data[i]=self.f(self.data[i*2],self.data[i*2+1])
  # 区間[l,r)へのクエリ
  def query(self,l,r):
    l,r=l+self.size,r+self.size
    lret,rret=self.default,self.default
    while l<r:
      if l&1:
        lret=self.f(self.data[l],lret)
        l+=1
      if r&1:
        r-=1
        rret=self.f(self.data[r],rret)
      l>>=1
      r>>=1
    return self.f(lret,rret)
  def get(self,i):
    return self.data[self.size+i]


def main1(n,xary,q,lrx):
  s=set(xary)
  for l,r,x in lrx:s.add(x)
  s=list(s)
  s.sort()
  numx=len(s)
  d={si:i for i,si in enumerate(s)}
  dr={i:si for i,si in enumerate(s)}
  didx={}
  for i,xv in enumerate(xary):
    xv=d[xv]
    if xv in didx:
      didx[xv].add(i)
    else:
      didx[xv]={i}
  del s
  inf=float('inf')
  mxst=SegmentTree(n,max,-inf)
  mnst=SegmentTree(n,min,inf)
  mnst.init_build([d[x] for x in xary])
  lrx=[[d[x],l,r,i] for i,(l,r,x) in enumerate(lrx)]
  lrx.sort()
  ret=[None]*q
  idx=0
  for x in range(numx):
    while idx<q and lrx[idx][0]==x:
      _,l,r,i=lrx[idx]
      y=mnst.query(l-1,r)
      z=mxst.query(l-1,r)
      tmp=inf
      if y!=inf:tmp=min(tmp,abs(dr[x]-dr[y]))
      if z!=-inf:tmp=min(tmp,abs(dr[x]-dr[z]))
      ret[i]=tmp
      idx+=1
      if idx==q:return ret
    if x not in didx:continue
    for i in didx[x]:
      mxst.update(i,x)
      mnst.update(i,inf)
  return ret

if __name__=='__main__':
  n=int(input())
  xary=list(map(int,input().split()))
  q=int(input())
  lrx=[list(map(int,input().split())) for _ in range(q)]
  ret1=main1(n,xary,q,lrx)
  print(*ret1,sep='\n')
0