結果

問題 No.875 Range Mindex Query
ユーザー Ueki
提出日時 2019-11-10 21:58:25
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 1,547 ms / 2,000 ms
コード長 2,315 bytes
コンパイル時間 362 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 28,600 KB
最終ジャッジ日時 2024-09-15 05:06:38
合計ジャッジ時間 11,966 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

# python template for atcoder1
import sys
sys.setrecursionlimit(10**9)
input = sys.stdin.readline


class SegmentTree():
    '''
    非再帰
    segment tree
    '''

    def __init__(self, n, func, init=float('inf')):
        '''
        n->配列の長さ
        func:func(a,b)->val,　func=minだとRMQになる
        木の高さhとすると,
        n:h-1までのノード数。h段目のノードにアクセスするために使う。
        data:ノード。
        parent:k->child k*2+1とk*2+2
        '''
        self.n = 2**(n-1).bit_length()
        self.init = init
        self.data = [init]*(2*self.n)
        self.func = func

    def set(self, k, v):
        '''
        あたいの初期化
        '''
        self.data[k+self.n-1] = v

    def swap(self, x, y):
        x_val, x_ind = self.data[x+self.n-1]
        y_val, y_ind = self.data[y+self.n-1]
        self.update(x, [y_val, x])
        self.update(y, [x_val, y])

    def build(self):
        '''
        setの後に一斉更新
        '''
        for k in reversed(range(self.n-1)):
            self.data[k] = self.func(self.data[k*2+1], self.data[k*2+2])

    def update(self, k, a):
        '''
        list[k]=aに更新する。
        更新ぶんをrootまで更新
        '''
        k += self.n-1
        self.data[k] = a

        while k > 0:
            k = (k-1)//2
            self.data[k] = self.func(self.data[k*2+1], self.data[k*2+2])

    def query(self, l, r):
        '''
        [l,r)のfuncを求める
        '''
        L = l+self.n
        R = r+self.n
        ret = self.init
        while L < R:
            if R & 1:
                R -= 1
                ret = self.func(ret, self.data[R-1])
            if L & 1:
                ret = self.func(ret, self.data[L-1])
                L += 1
            L >>= 1
            R >>= 1
        return ret


N, Q = map(int, input().split())
L = list(map(int, input().split()))


def f(a, b):
    if a[0] > b[0]:
        return b
    else:
        return a


Seg = SegmentTree(N, f, init=[float('inf'), -1])
for i, l in enumerate(L):
    Seg.set(i, [l, i])
Seg.build()

for _ in range(Q):
    q, l, r = map(lambda x: int(x)-1, input().split())
    if q == 0:
        Seg.swap(l, r)
    else:
        val, ind = Seg.query(l, r+1)
        print(ind+1)
0