結果
| 問題 | 
                            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 | 
ソースコード
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()
            
            
            
        
            
brthyyjp