結果

問題 No.1332 Range Nearest Query
コンテスト
ユーザー titia
提出日時 2021-01-14 05:33:29
言語 PyPy3
(7.3.17)
結果
WA  
実行時間 -
コード長 3,346 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 203 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 128,176 KB
最終ジャッジ日時 2024-11-23 16:47:42
合計ジャッジ時間 20,979 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 19 WA * 29
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = sys.stdin.readline

N=int(input())
A=list(map(int,input().split()))
Q=int(input())
Query=[tuple(map(int,input().split())) for i in range(Q)]

def seg_function1(x,y): # Segment treeで扱うfunction
    return max(x,y)

seg_el=1<<(N.bit_length()) # Segment treeの台の要素数
SEG1=[0]*(2*seg_el) # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化

for i in range(N): # Aを対応する箇所へupdate
    SEG1[i+seg_el]=A[i]

for i in range(seg_el-1,0,-1): # 親の部分もupdate
    SEG1[i]=seg_function1(SEG1[i*2],SEG1[i*2+1])

def update1(n,x,seg_el): # A[n]をxへ更新（反映）
    i=n+seg_el
    SEG1[i]=x
    i>>=1 # 子ノードへ
    
    while i!=0:
        SEG1[i]=seg_function(SEG1[i*2],SEG1[i*2+1])
        i>>=1
        
def getvalues1(l,r): # 区間[l,r)に関するseg_functionを調べる
    L=l+seg_el
    R=r+seg_el
    ANS=0

    while L<R:
        if L & 1:
            ANS=seg_function1(ANS , SEG1[L])
            L+=1

        if R & 1:
            R-=1
            ANS=seg_function1(ANS , SEG1[R])
        L>>=1
        R>>=1

    return ANS

def bisect_on_SEG1(l,x): # l以上で、x以上の最小のindexを返す。ただし、存在しない場合はNを返す。
    L=l+seg_el
    R=seg_el*2-1
    while L<R:
        if SEG1[L]>=x:
            break
        if L & 1:
            L+=1
        L>>=1
        R>>=1
 
    if SEG1[L]<x:
        return N
 
    while L<seg_el:
        if SEG1[L*2]>=x:
            L=L*2
        else:
            L=L*2+1
 
    return L-seg_el

def seg_function2(x,y): # Segment treeで扱うfunction
    return min(x,y)

seg_el=1<<(N.bit_length()) # Segment treeの台の要素数
SEG2=[1<<30]*(2*seg_el) # 1-indexedなので、要素数2*seg_el.Segment treeの初期値で初期化

for i in range(N): # Aを対応する箇所へupdate
    SEG2[i+seg_el]=A[i]

for i in range(seg_el-1,0,-1): # 親の部分もupdate
    SEG2[i]=seg_function2(SEG2[i*2],SEG2[i*2+1])

def update2(n,x,seg_el): # A[n]をxへ更新（反映）
    i=n+seg_el
    SEG2[i]=x
    i>>=1 # 子ノードへ
    
    while i!=0:
        SEG2[i]=seg_function2(SEG2[i*2],SEG2[i*2+1])
        i>>=1
        
def getvalues2(l,r): # 区間[l,r)に関するseg_functionを調べる
    L=l+seg_el
    R=r+seg_el
    ANS=1<<30

    while L<R:
        if L & 1:
            ANS=seg_function2(ANS , SEG2[L])
            L+=1

        if R & 1:
            R-=1
            ANS=seg_function2(ANS , SEG2[R])
        L>>=1
        R>>=1

    return ANS

def bisect_on_SEG2(l,x): # l以上で、x以下の最小のindexを返す。ただし、存在しない場合はNを返す。
    L=l+seg_el
    R=seg_el*2-1
    while L<R:
        if SEG2[L]<=x:
            break
        if L & 1:
            L+=1
        L>>=1
        R>>=1
 
    if SEG2[L]>x:
        return N
 
    while L<seg_el:
        if SEG2[L*2]<=x:
            L=L*2
        else:
            L=L*2+1
 
    return L-seg_el

for l,r,x in Query:
    xx=bisect_on_SEG1(l-1,x)
    yy=bisect_on_SEG2(l-1,x)

    #print(xx,yy)
    
    ANS=1<<30
    if l-1<=xx-1<=r-1:
        ANS=min(ANS,abs(getvalues1(l-1,xx-1)-x))
    if l-1<=yy-1<=r-1:
        ANS=min(ANS,abs(getvalues2(l-1,yy-1)-x))

    for i in [l-1,r-1,xx-1,xx,yy-1,yy]:
        if l-1<=i<=r-1:
            ANS=min(ANS,abs(A[i]-x))

    print(ANS)
0