結果

問題 No.1536 仕切り直し
ユーザー qib
提出日時 2023-03-03 19:03:52
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 300 ms / 2,000 ms
コード長 1,049 bytes
コンパイル時間 186 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 122,436 KB
最終ジャッジ日時 2024-09-17 22:12:26
合計ジャッジ時間 2,881 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

INF = 1 << 60

n, m = map(int, input().split())
a = [0] + list(map(int, input().split()))

dpmin = [[INF for _ in range(m + 1)] for _ in range(n + 1)]
dpmax = [[- INF for _ in range(m + 1)] for _ in range(n + 1)]
dpmin[-1][0] = a[-1]
dpmax[-1][0] = a[-1]
for i in range(n, 0, -1):
  for j in range(m + 1):
    if j < m:
      dpmin[i - 1][j + 1] = min(dpmin[i - 1][j + 1], - dpmax[i][j] + a[i - 1])
      dpmax[i - 1][j + 1] = max(dpmax[i - 1][j + 1], - dpmin[i][j] + a[i - 1])
    dpmin[i - 1][j] = min(dpmin[i - 1][j], dpmin[i][j] + a[i - 1])
    dpmax[i - 1][j] = max(dpmax[i - 1][j], dpmax[i][j] + a[i - 1])

rest = 0
for j in range(m + 1):
  if dpmax[0][rest] < dpmax[0][j]:
    rest = j

ans = []
flg = False
for i in range(n):
  if rest <= 0:
    break

  if flg:
    if dpmin[i][rest] != dpmin[i + 1][rest] + a[i]:
      ans.append(i)
      flg = False
      rest -= 1
  else:
    if dpmax[i][rest] != dpmax[i + 1][rest] + a[i]:
      ans.append(i)
      flg = True
      rest -= 1

while len(ans) < m:
  ans.append(n)

print(*ans, sep="\n")
0