結果
問題 |
No.1391 ±1 Abs Sum
|
ユーザー |
![]() |
提出日時 | 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 |
ソースコード
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))