結果

問題 No.1606 Stuffed Animals Keeper
コンテスト
ユーザー あかしあみどり
提出日時 2023-03-01 22:21:20
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,601 bytes
コンパイル時間 550 ms
コンパイル使用メモリ 82,048 KB
実行使用メモリ 201,728 KB
最終ジャッジ日時 2024-09-16 19:26:59
合計ジャッジ時間 8,632 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 16 WA * 32
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

def oi(): return int(input())
def os(): return input()
def mi(): return list(map(int, input().split()))

# import sys
# input = sys.stdin.readline
# import sys
# sys.setrecursionlimit(10**8)
# import pypyjit
# pypyjit.set_param('max_unroll_recursion=-1')
input_count = 0

from collections import Counter
N = oi()
A = mi()
elements = []
temp = {0:0, 1:0}
for a in A:
    if a == 2:
        if len(temp)!=0:
            elements.append(temp)
        temp = {0:0, 1:0}
    else:
        temp[a] += 1            
if len(temp)!=0:
    elements.append(temp)
zero_num = 0
one_num = 0
diffs = []
for e in elements:
    if e[0] > e[1]:
        one_num += e[1]
        diffs.append([e[0] + e[1], e[0] - e[1], 0]) #  全体の0と1の, 増える交換回数, 同じかフラグ
    elif e[0] < e[1]:
        zero_num += e[0]
        diffs.append([-1*(e[0] + e[1]), e[1] - e[0], 0])
    else:
        one_num += e[1]
        diffs.append([e[0] + e[1], 0, 1])
if zero_num ==  one_num:
    print(zero_num)
else:
    # 一致してなければDP
    INF = 10**5
    dp = [[INF] * (N*2 +2) for _ in range(len(elements)+1)]

    dp[0][N+1] = 0
    for y,d in enumerate(diffs):
        for x in range(0, 2*N+2):
            if dp[y][x] != INF:
                if 0 <= x+d[0] <= 2*N+1:
                    dp[y+1][x+d[0]] = min(dp[y+1][x+d[0]], dp[y][x]+d[1])
                if d[2] == 1:
                    if 0 <= x-d[0] <= 2*N+1:
                        dp[y+1][x-d[0]] = min(dp[y+1][x-d[0]], dp[y][x]+d[1])

    ret = dp[-1][zero_num-one_num + N+1]
    if ret != INF:
        print(ret)
    else:
        print(-1)
0