結果

問題 No.3266 岩井星人は見ずにはいられない
コンテスト
ユーザー iastm
提出日時 2025-09-06 15:49:10
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 66 ms / 2,000 ms
コード長 962 bytes
コンパイル時間 290 ms
コンパイル使用メモリ 82,748 KB
実行使用メモリ 66,900 KB
最終ジャッジ日時 2025-09-06 15:49:13
合計ジャッジ時間 3,065 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

N, A = (int(x) for x in input().split())
S = input()

result = 0
r = 1200
nr = r
count = 0
for i in range(N):
    if S[i] == '0':
        nr -= 1
    elif nr < 1200:
        nr += 1
        count += 1
    if count == A:
        print(i + 1)
        exit()

result += N
A -= count

while True:
    nnr = nr
    ncount = 0
    for i in range(N):
        if S[i] == '0':
            nnr -= 1
        elif nnr < 1200:
            nnr += 1
            ncount += 1
        if ncount == A:
            print(result + i + 1)
            exit()

    result += N
    A -= ncount

    if nnr - nr != nr - r:
        r = nr
        nr = nnr
        continue

    result += ((A - 1) // ncount) * N
    A %= ncount
    if A == 0:
        A = ncount
    count = 0
    for i in range(N):
        if S[i] == '0':
            nnr -= 1
        elif nnr < 1200:
            nnr += 1
            count += 1
        if count == A:
            print(result + i + 1)
            exit()
0