結果

問題 No.3266 岩井星人は見ずにはいられない
ユーザー detteiuu
提出日時 2025-09-06 14:54:50
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 186 ms / 2,000 ms
コード長 943 bytes
コンパイル時間 181 ms
コンパイル使用メモリ 82,460 KB
実行使用メモリ 156,840 KB
最終ジャッジ日時 2025-09-06 14:54:56
合計ジャッジ時間 4,381 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

MAX = 0
cnt = 0
for s in S:
    if s == "0":
        cnt -= 1
    else:
        cnt += 1
    MAX = max(MAX, cnt)
limit = 1200-MAX

cnt1 = S.count("1")
visited = set()
rate = 1200
now = 0
ans = 0
counter = 0
while rate not in visited:
    visited.add(rate)
    c = max(rate-limit, 0)
    if A <= now+(cnt1-c):
        break
    ans += N
    now += cnt1-c
    rate += cnt1-(N-cnt1)-c
    counter += 1
    if counter == 10**5*5:
        break
if rate in visited:
    c = max(rate-limit, 0)
    cnt = (A-now-1)//(cnt1-c)
    ans += N*cnt
    now += (cnt1-c)*cnt
    rate += (cnt1-(N-cnt1)-c)*cnt
elif now+(cnt1-c) < A:
    cnt = (A-now-1)//cnt1
    ans += N*cnt
    now += cnt1*cnt
    rate += (cnt1-(N-cnt1))*cnt

for s in S:
    ans += 1
    if s == "0":
        rate -= 1
    else:
        if rate < 1200:
            rate += 1
            now += 1
    if now == A:
        print(ans)
        break
0