結果

問題 No.3266 岩井星人は見ずにはいられない
ユーザー iastm
提出日時 2025-09-06 15:45:00
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 927 bytes
コンパイル時間 319 ms
コンパイル使用メモリ 82,108 KB
実行使用メモリ 68,176 KB
最終ジャッジ日時 2025-09-06 15:45:09
合計ジャッジ時間 3,310 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 3
other WA * 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:
            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