結果

問題 No.1558 Derby Live
ユーザー taiga0629kyoprotaiga0629kyopro
提出日時 2021-06-26 01:34:43
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 2,861 bytes
コンパイル時間 254 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 109,232 KB
最終ジャッジ日時 2024-06-25 09:04:56
合計ジャッジ時間 19,373 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 477 ms
86,200 KB
testcase_01 WA -
testcase_02 AC 39 ms
52,676 KB
testcase_03 WA -
testcase_04 WA -
testcase_05 AC 393 ms
85,592 KB
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 AC 39 ms
53,292 KB
testcase_34 AC 39 ms
52,968 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

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

#####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:
            self.tree[k >> 1] = self.segfunc(self.tree[k ^ 1],self.tree[k])
            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