結果

問題 No.1536 仕切り直し
ユーザー qibqib
提出日時 2023-03-03 19:03:52
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 302 ms / 2,000 ms
コード長 1,049 bytes
コンパイル時間 343 ms
コンパイル使用メモリ 81,636 KB
実行使用メモリ 121,988 KB
最終ジャッジ日時 2023-10-18 01:12:50
合計ジャッジ時間 4,003 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 39 ms
53,376 KB
testcase_01 AC 39 ms
53,376 KB
testcase_02 AC 38 ms
53,376 KB
testcase_03 AC 74 ms
72,364 KB
testcase_04 AC 144 ms
88,024 KB
testcase_05 AC 233 ms
106,948 KB
testcase_06 AC 77 ms
74,856 KB
testcase_07 AC 77 ms
74,268 KB
testcase_08 AC 129 ms
86,368 KB
testcase_09 AC 162 ms
91,720 KB
testcase_10 AC 302 ms
121,988 KB
testcase_11 AC 116 ms
83,148 KB
testcase_12 AC 234 ms
107,604 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