結果

問題 No.1558 Derby Live
ユーザー taiga0629kyoprotaiga0629kyopro
提出日時 2021-06-26 01:36:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 692 ms / 2,000 ms
コード長 2,951 bytes
コンパイル時間 370 ms
コンパイル使用メモリ 87,300 KB
実行使用メモリ 109,848 KB
最終ジャッジ日時 2023-09-07 15:16:12
合計ジャッジ時間 22,268 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 504 ms
88,140 KB
testcase_01 AC 605 ms
109,848 KB
testcase_02 AC 69 ms
71,568 KB
testcase_03 AC 547 ms
87,996 KB
testcase_04 AC 385 ms
86,396 KB
testcase_05 AC 439 ms
87,300 KB
testcase_06 AC 511 ms
93,096 KB
testcase_07 AC 125 ms
79,508 KB
testcase_08 AC 508 ms
92,768 KB
testcase_09 AC 583 ms
96,888 KB
testcase_10 AC 566 ms
97,552 KB
testcase_11 AC 576 ms
96,444 KB
testcase_12 AC 562 ms
97,836 KB
testcase_13 AC 578 ms
97,840 KB
testcase_14 AC 577 ms
100,456 KB
testcase_15 AC 585 ms
99,352 KB
testcase_16 AC 604 ms
101,444 KB
testcase_17 AC 608 ms
102,096 KB
testcase_18 AC 615 ms
102,196 KB
testcase_19 AC 622 ms
103,836 KB
testcase_20 AC 651 ms
104,064 KB
testcase_21 AC 607 ms
103,044 KB
testcase_22 AC 613 ms
104,424 KB
testcase_23 AC 625 ms
104,100 KB
testcase_24 AC 636 ms
103,628 KB
testcase_25 AC 692 ms
104,412 KB
testcase_26 AC 632 ms
105,008 KB
testcase_27 AC 624 ms
102,824 KB
testcase_28 AC 628 ms
103,108 KB
testcase_29 AC 639 ms
103,848 KB
testcase_30 AC 627 ms
103,096 KB
testcase_31 AC 638 ms
103,372 KB
testcase_32 AC 664 ms
105,052 KB
testcase_33 AC 71 ms
71,396 KB
testcase_34 AC 72 ms
71,496 KB
権限があれば一括ダウンロードができます

ソースコード

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