結果

問題 No.1536 仕切り直し
ユーザー qibqib
提出日時 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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
51,712 KB
testcase_01 AC 38 ms
51,584 KB
testcase_02 AC 37 ms
52,480 KB
testcase_03 AC 71 ms
72,320 KB
testcase_04 AC 131 ms
88,576 KB
testcase_05 AC 234 ms
107,264 KB
testcase_06 AC 80 ms
77,396 KB
testcase_07 AC 74 ms
74,624 KB
testcase_08 AC 125 ms
86,528 KB
testcase_09 AC 159 ms
92,596 KB
testcase_10 AC 300 ms
122,436 KB
testcase_11 AC 116 ms
83,456 KB
testcase_12 AC 231 ms
107,776 KB
権限があれば一括ダウンロードができます

ソースコード

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