結果

問題 No.1558 Derby Live
コンテスト
ユーザー aaaaaaaaaa2230
提出日時 2021-06-26 17:17:50
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,622 bytes
コンパイル時間 288 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 115,328 KB
最終ジャッジ日時 2024-06-25 09:10:57
合計ジャッジ時間 18,259 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample WA * 2
other AC * 1 WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff # 

class SegTree:
    """ define what you want to do with 0 index, ex) size = tree_size, func = min or max, sta = default_value """
    
    def __init__(self,size,func,sta):
        self.n = size
        self.size = 1 << size.bit_length()
        self.func = func
        self.sta = sta
        self.tree = [sta]*(2*self.size)

    def set(self,i,x):
        i += self.size
        self.tree[i] = x
        while i > 1:
            i >>= 1
            self.tree[i] = self.func(self.tree[i<<1],self.tree[i<<1 | 1])

    
    def get(self,l,r):
        """ take the value of [l r) with func (min or max)"""
        l += self.size
        r += self.size
        res = self.sta

        while l < r:
            if l & 1:
                res = self.func(self.tree[l],res)
                l += 1
            if r & 1:
                res = self.func(self.tree[r-1],res)
            l >>= 1
            r >>= 1
        return res
        

n,m,q = map(int,input().split())
Q = [list(map(int,input().split())) for i in range(q)]

def func(a,b):
    return [b[aa] for aa in a]
seg = SegTree(m,func,[i for i in range(n)])

def sol(A):
    ans = [0]*n
    for i,a in enumerate(A):
        ans[a] = i+1
    return ans
for q in Q:
    id = q[0]
    if id == 1:
        d = q[1]
        p = [x-1 for x in q[2:]]
        seg.set(d-1,p)

    elif id == 2:
        print(*sol(seg.get(0,q[1])))
    
    else:
        l,r = q[1:]
        a = seg.get(0,l)
        b = seg.get(0,r)
        ans = 0
        for i in range(n):
            for j in range(n):
                if a[i] == b[j]:
                    ans += abs(i-j)
        print(ans)
0