結果

問題 No.1077 Noelちゃんと星々4
ユーザー brthyyjp
提出日時 2020-06-12 21:35:41
言語 PyPy3
(7.3.15)
結果
TLE  
実行時間 -
コード長 1,468 bytes
コンパイル時間 193 ms
コンパイル使用メモリ 82,664 KB
実行使用メモリ 148,204 KB
最終ジャッジ日時 2024-06-24 04:34:50
合計ジャッジ時間 30,843 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 14 TLE * 6
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input = sys.stdin.buffer.readline

def main():

    N = int(input())
    Y = list(map(int, input().split()))

    n = 10**4+1

    def segfunc(x, y):
        return min(x, y)

    def init(init_val):
        # set_val
        for i in range(n):
            seg[i+num-1] = init_val[i]
        # built
        for i in range(num-2, -1, -1):
            seg[i] = segfunc(seg[2*i+1], seg[2*i+2])

    def update(k, x):
        k += num - 1
        seg[k] = x
        while k:
            k = (k-1)//2
            seg[k] = segfunc(seg[2*k+1], seg[2*k+2])

    def query(p, q):
        if q <= p:
            return ide_ele
        p += num - 1
        q += num - 2
        res = ide_ele
        while q-p>1:
            if p&1 == 0:
                res = segfunc(res, seg[p])
            if q&1 == 1:
                res = segfunc(res, seg[q])
                q -= 1
            p = p//2
            q = (q-1)//2
        if p == q:
            res = segfunc(res, seg[p])
        else:
            res = segfunc(segfunc(res, seg[p]), seg[q])
        return res

    # identity element
    ide_ele = 10**18

    # num: n以上の最小の2のべき乗
    num = 2**(n-1).bit_length()
    seg = [ide_ele]*2*num

    dp = [0]*n
    import copy
    for i in range(N):
        init(dp)
        temp = [0]*n
        for j in range(n):
            temp[j] = query(0, j+1)+abs(j-Y[i])
        dp = copy.copy(temp)
    print(min(dp))

if __name__ =='__main__':
    main()
0