結果

問題 No.1391 ±1 Abs Sum
ユーザー tpynerivertpyneriver
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,264 KB
testcase_01 AC 38 ms
52,616 KB
testcase_02 AC 38 ms
52,284 KB
testcase_03 AC 38 ms
53,760 KB
testcase_04 AC 39 ms
52,932 KB
testcase_05 AC 37 ms
53,504 KB
testcase_06 AC 40 ms
53,556 KB
testcase_07 AC 38 ms
52,820 KB
testcase_08 AC 40 ms
53,320 KB
testcase_09 AC 40 ms
53,212 KB
testcase_10 AC 40 ms
53,804 KB
testcase_11 AC 232 ms
108,964 KB
testcase_12 AC 188 ms
104,932 KB
testcase_13 AC 212 ms
107,116 KB
testcase_14 AC 237 ms
98,620 KB
testcase_15 AC 242 ms
100,108 KB
testcase_16 AC 271 ms
104,900 KB
testcase_17 AC 249 ms
104,124 KB
testcase_18 AC 271 ms
109,516 KB
testcase_19 AC 179 ms
103,368 KB
testcase_20 AC 240 ms
108,864 KB
testcase_21 AC 171 ms
104,548 KB
testcase_22 AC 164 ms
105,168 KB
testcase_23 AC 175 ms
109,292 KB
testcase_24 AC 171 ms
108,468 KB
testcase_25 AC 173 ms
108,436 KB
testcase_26 AC 172 ms
108,880 KB
testcase_27 AC 165 ms
101,116 KB
testcase_28 AC 166 ms
100,912 KB
testcase_29 AC 174 ms
108,512 KB
testcase_30 AC 181 ms
104,588 KB
testcase_31 AC 245 ms
108,948 KB
testcase_32 AC 127 ms
96,436 KB
testcase_33 AC 142 ms
101,444 KB
testcase_34 AC 135 ms
108,544 KB
testcase_35 AC 37 ms
53,932 KB
testcase_36 AC 139 ms
109,372 KB
権限があれば一括ダウンロードができます

ソースコード

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