結果

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

ソースコード

diff # 

def main0(n,xary,q,lrx):
  ret=[]
  inf=float('inf')
  for l,r,x in lrx:
    tmp=inf
    for y in range(l-1,r):
      tmp=min(tmp,abs(xary[y]-x))
    ret.append(tmp)
  return ret

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]



n=int(input())
xary=list(map(int,input().split()))
q=int(input())
lrx=[list(map(int,input().split())) for _ in range(q)]
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:break
  if x not in didx:continue
  for i in didx[x]:
    mxst.update(i,x)
    mnst.update(i,inf)
print(*ret,sep='\n')
0