結果

問題 No.1391 ±1 Abs Sum
ユーザー tpyneriver
提出日時 2021-02-13 12:49:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 271 ms / 2,000 ms
コード長 998 bytes
コンパイル時間 208 ms
コンパイル使用メモリ 82,080 KB
実行使用メモリ 109,516 KB
最終ジャッジ日時 2024-07-20 15:46:42
合計ジャッジ時間 7,168 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #

import sys
readline = sys.stdin.readline

def monotoneminima(f, N, X):
    midx = [None]*N
    mval = [None]*N
    
    stack = [(0, N, 0, X)]
    while stack:
        u, d, l, r = stack.pop()
        
        m = (u+d)//2
        av = f(m, l)
        ai = l  
        for i in range(l+1, r):
            fv = f(m, i)
            if fv < av:
                ai = i
                av = fv
        midx[m] = ai
        mval[m] = av
        if u != m and m != d:
            stack.append((u, m, l, ai+1))
        if m+1 != d and u != m+1:
            stack.append((m+1, d, ai, r))
    return midx, mval

def calc(i, j):
    return -g(i, N-1) + 2*g(i, j+K-1) - 2*g(i, j-1)
def g(i, j):
    if j == -1:
        return 0
    if i <= j:
        return A[i]*(2*i-j+1) + S[j] - 2*S[i]
    return A[i]*(j+1) - S[j]


N, K = map(int, readline().split())
A = list(map(int, readline().split()))
S = A[:] + [0]
for i in range(1, N):
    S[i] += S[i-1]
    
mi, mv = monotoneminima(calc, N, N-K+1)
print(min(mv))
0