結果

問題 No.78 クジ付きアイスバー
ユーザー yn
提出日時 2015-06-25 03:17:07
言語 Python3
(3.13.1 + numpy 2.2.1 + scipy 1.14.1)
結果
AC  
実行時間 31 ms / 5,000 ms
コード長 2,003 bytes
コンパイル時間 205 ms
コンパイル使用メモリ 12,800 KB
実行使用メモリ 10,880 KB
最終ジャッジ日時 2024-11-08 00:33:53
合計ジャッジ時間 2,253 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

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
    mustend = [False] * l
    for i in range(l):
        if S[i] == '0':
            if atariend > 0:
                atariend -= 1
            else:
                mustend[i] = True

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

    countend = 0

    for i in range(l):
        if mustend[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