結果

問題 No.875 Range Mindex Query
ユーザー AEn
提出日時 2022-06-16 23:13:27
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 571 ms / 2,000 ms
コード長 2,198 bytes
コンパイル時間 215 ms
コンパイル使用メモリ 82,260 KB
実行使用メモリ 116,364 KB
最終ジャッジ日時 2024-10-07 07:51:17
合計ジャッジ時間 6,108 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 18
権限があれば一括ダウンロードができます

ソースコード

diff #

from collections import defaultdict
class SegmentTree():
    """UnitXは単位元、fは区間で行いたい操作、initは自然数あるいは配列"""
    def __init__(self, init, unitX, f):
        self.f = f # (X, X) -> X
        self.unitX = unitX
        self.f = f
        if type(init) == int:
            self.n = init
            self.n = 1 << (self.n - 1).bit_length()
            self.X = [unitX] * (self.n * 2)
        else:
            self.n = len(init)
            self.n = 1 << (self.n - 1).bit_length()
            # len(init)が2の累乗ではない時UnitXで埋める
            self.X = [unitX] * self.n + init + [unitX] * (self.n - len(init))
            # 配列のindex1まで埋める
            for i in range(self.n-1, 0, -1):
                self.X[i] = self.f(self.X[i*2], self.X[i*2|1])
    
    # 0-indexedのi番目の値をxで置換
    def update(self, i, x):
        # 最下段に移動
        i += self.n
        self.X[i] = x
        # 上向に更新
        i >>= 1
        while i:
            self.X[i] = self.f(self.X[i*2], self.X[i*2|1])
            i >>= 1
    
    # 元の配列のindexの値を見る
    def getvalue(self, i):
        return self.X[i + self.n]
    
    # 区間[l, r)でのfを行った値
    def getrange(self, l, r):
        l += self.n
        r += self.n
        al = self.unitX
        ar = self.unitX
        while l < r:
            # 左端が右子ノードであれば
            if l & 1:
                al = self.f(al, self.X[l])
                l += 1
            # 右端が右子ノードであれば
            if r & 1:
                r -= 1
                ar = self.f(self.X[r], ar)
            l >>= 1
            r >>= 1
        return self.f(al, ar)
    
 
N, Q = map(int, input().split())
a = list(map(int, input().split()))
d = defaultdict(int)
for i in range(N):
    d[a[i]] = i+1
st = SegmentTree(a, float('inf'), min)
for i in range(Q):
    p, l, r = map(int, input().split())
    if p == 1:
        x, y = st.getvalue(l-1),st.getvalue(r-1)
        d[x], d[y] = d[y], d[x]
        st.update(l-1, y)
        st.update(r-1, x)
    else:
        res = st.getrange(l-1, r)
        print(d[res])
0