結果

問題 No.1558 Derby Live
コンテスト
ユーザー taiga0629kyopro
提出日時 2021-06-26 01:36:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 656 ms / 2,000 ms
コード長 2,951 bytes
コンパイル時間 285 ms
コンパイル使用メモリ 82,248 KB
実行使用メモリ 109,336 KB
最終ジャッジ日時 2024-06-25 09:05:21
合計ジャッジ時間 21,118 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff # 

n,m,q=map(int,input().split())
#１点更新、区間取得

#####segfunc#####
def segfunc(x, y):
    res=[0]*(n+1)
    for i in range(1,n+1):
        res[i]=y[x[i]]
    return res
#################
add=segfunc
a1=[0,1,3,2]
a2=[0,2,1,3]
a3=[0,2,3,1]


#####ide_ele#####
ide_ele =[i for i in range(n+1)]
#################


class SegTree:
    """
    init(init_val, ide_ele): 配列init_valで初期化 O(N)
    update(k, x): k番目の値をxに更新 O(logN)
    query(l, r): 区間[l, r)をsegfuncしたものを返す O(logN)
    """
    def __init__(self, init_val, segfunc, ide_ele):
        """
        init_val: 配列の初期値
        segfunc: 区間にしたい操作
        ide_ele: 単位元
        n: 要素数
        num: n以上の最小の2のべき乗
        tree: セグメント木(1-index)
        """
        n = len(init_val)
        self.segfunc = segfunc
        self.ide_ele = ide_ele
        self.num = 1 << (n - 1).bit_length()
        self.tree = [ide_ele] * 2 * self.num
        # 配列の値を葉にセット
        for i in range(n):
            self.tree[self.num + i] = init_val[i]
        # 構築していく
        for i in range(self.num - 1, 0, -1):
            self.tree[i] = self.segfunc(self.tree[2 * i], self.tree[2 * i + 1])

    def update(self, k, x):
        """
        k番目の値をxに更新
        k: index(0-index)
        x: update value
        """
        k += self.num
        self.tree[k] = x
        while k > 1:
            if k%2==1:self.tree[k >> 1] = self.segfunc(self.tree[k ^ 1],self.tree[k])
            else:self.tree[k >> 1] = self.segfunc(self.tree[k],self.tree[k ^ 1])
            k >>= 1

    def query(self, l, r):
        """
        [l, r)のsegfuncしたものを得る
        l: index(0-index)
        r: index(0-index)
        """
        sml = self.ide_ele
        smr = self.ide_ele
        l += self.num
        r += self.num
        while (l < r):
            if (l & 1):
                sml = self.segfunc(sml, self.tree[l])
                l += 1
            if (r & 1):
                smr = self.segfunc(self.tree[r - 1], smr)
                r -= 1
            l >>= 1
            r >>= 1
        return self.segfunc(sml, smr)

    def get(self,i):
        """
        i番目の値を得る
        """
        return self.query(i,i+1)

    def add(self,i,x):
        """
        i番目にxをたす
        """
        self.update(i,self.get(i)+x)


seg=SegTree([ide_ele for _ in range(m+3)],segfunc,ide_ele)
for _ in range(q):
    que=list(map(int,input().split()))
    if que[0]==1:
        p=[0]+que[2:]
        seg.update(que[1],p)
    elif que[0]==2:
        p=seg.query(0,que[1]+1)
        res=[0]*(n+1)
        for i in range(1,n+1):
            res[p[i]]=i
        print(*res[1:])

    else:
        l,r=que[1],que[2]
        p=seg.query(l,r+1)
        res=0
        for i in range(1,n+1):
            res+=abs(p[i]-i)
        print(res)
0