結果

問題 No.1606 Stuffed Animals Keeper
ユーザー brthyyjp
提出日時 2022-01-11 14:11:35
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 85 ms / 3,000 ms
コード長 553 bytes
コンパイル時間 1,220 ms
コンパイル使用メモリ 82,400 KB
実行使用メモリ 76,660 KB
最終ジャッジ日時 2024-11-14 11:38:26
合計ジャッジ時間 5,605 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff #

n = int(input())
A = list(map(int, input().split()))
A += [2]
s = 0
for a in A:
    s += (a == 0)
C = []
cnt0 = 0
cnt1 = 0
for i, a in enumerate(A):
    if a == 2:
        if cnt0 or cnt1:
            C.append((cnt0, cnt1))
            cnt0 = 0
            cnt1 = 0
    elif a == 0:
        cnt0 += 1
    else:
        cnt1 += 1

INF = 10**18
dp = [INF]*(s+1)
dp[0] = 0
for c0, c1 in C:
    for j in reversed(range(s+1)):
        if j+c0+c1 <= s:
            dp[j+c0+c1] = min(dp[j+c0+c1], dp[j]+c1)
if dp[s] < INF:
    print(dp[s])
else:
    print(-1)
0