結果

問題 No.78 クジ付きアイスバー
ユーザー yn
提出日時 2015-06-25 02:43:15
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
WA  
実行時間 -
コード長 1,839 bytes
コンパイル時間 279 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-10-06 17:10:13
合計ジャッジ時間 2,536 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other AC * 19 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, K = map(int,input().split())
S = input()

must = [False] * N
atari = 0
buy = 0
if N > K :
    for i in range(N):
        if S[i] == '0':
            if atari > 0:
                atari -= 1
            else:
                buy += 1

        elif S[i] == '1':
            if atari > 0 :
                atari -= 1
            else:
                buy += 1
            atari += 1
        
        else :
            if atari > 0:
                atari -= 1
            else:
                buy += 1
            atari += 2

        if i+1 == K:
            break

else:
    for i in range(N):
        if S[i] == '0':
            if atari > 0:
                atari -= 1
            else:
                must[i] = True

        elif S[i] == '1':
            if atari > 0 :
                atari -= 1
            else:
                must[i] = True
            atari += 1
        
        else :
            if atari > 0:
                atari -= 1
            else:
                must[i] = True
            atari += 2

    count  = 0
    for i in range(N):
        if must[i]:
            count += 1
            
    l = K % N

    atariend = atari

    for i in range(l):
        if S[i] == '0':
            if atariend > 0:
                atariend -= 1

        elif S[i] == '1':
            if atariend > 0 :
                atariend -= 1
            atariend += 1
        
        else :
            if atariend > 0:
                atariend -= 1

        countend = 0
        for i in range(l):
            if must[i]:
                countend += 1

        if atari >= count :
            buy = count
        else:
            if l == 0:
                buy = count + (count - atari) * (K // N - 1)
            
            else :
                buy = count + (count - atari) * (K // N - 1) + countend
            
print(buy)
0