結果

問題 No.1606 Stuffed Animals Keeper
コンテスト
ユーザー tamato
提出日時 2021-07-16 22:52:23
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 86 ms / 3,000 ms
コード長 850 bytes
コンパイル時間 144 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 76,928 KB
最終ジャッジ日時 2024-07-06 10:23:49
合計ジャッジ時間 4,152 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 48
権限があれば一括ダウンロードができます

ソースコード

diff # 

mod = 1000000007
eps = 10**-9
inf = 10**15


def main():
    import sys
    input = sys.stdin.readline

    N = int(input())
    A = list(map(int, input().split()))
    A.append(2)

    B = []
    cnt = [0, 0]
    for a in A:
        if a == 2:
            if sum(cnt):
                B.append(cnt)
                cnt = [0, 0]
        else:
            cnt[a] += 1

    S = 0
    for a, b in B:
        S += b

    dp = [inf] * (S+1)
    dp[S] = 0
    for i in range(len(B)):
        n0, n1 = B[i]
        dp_new = [inf] * (S+1)
        for j in range(S+1):
            dp_new[j] = min(dp_new[j], dp[j])
            if j - (n0 + n1) >= 0:
                dp_new[j - (n0 + n1)] = min(dp_new[j - (n0 + n1)], dp[j] + n0)
        dp = dp_new
    if dp[0] != inf:
        print(dp[0])
    else:
        print(-1)


if __name__ == '__main__':
    main()
0