結果
問題 | No.1606 Stuffed Animals Keeper |
ユーザー |
👑 ![]() |
提出日時 | 2021-04-15 21:43:58 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 73 ms / 3,000 ms |
コード長 | 1,028 bytes |
コンパイル時間 | 267 ms |
コンパイル使用メモリ | 82,436 KB |
実行使用メモリ | 70,748 KB |
最終ジャッジ日時 | 2024-07-06 07:49:33 |
合計ジャッジ時間 | 4,037 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 48 |
ソースコード
import sysfrom sys import stdinN = int(stdin.readline())A = list(map(int,stdin.readline().split()))#制約チェック#assert 2 <= N <= 3000#assert len(A) == N#for i in A:# assert 0 <= i <= 2lis = [] #2で区切られた各区間に関して (0の数,1の数) を入れていくallzero = 0 #0の総数allone = 0 #1の総数zero = 0one = 0for i in A:if i == 0:zero += 1allzero += 1elif i == 1:one += 1allone += 1elif zero + one > 0:lis.append( (zero,one) )zero = 0one = 0if zero + one > 0:lis.append( (zero,one) )M = allzero + allone#ちょうど allzero 個入る区間の組み合わせのうち、#oneの数の総和が最小の区間の組み合わせを求めればよいdp = [float("inf")] * (allzero + 1)dp[0] = 0for nz,no in lis:for i in range(allzero-(nz+no),-1,-1):dp[i + (nz+no)] = min(dp[i + (nz+no)] , dp[i] + no)if dp[-1] == float("inf"):print (-1)else:print (dp[-1])