結果
問題 |
No.3266 岩井星人は見ずにはいられない
|
ユーザー |
![]() |
提出日時 | 2025-09-06 14:22:02 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 69 ms / 2,000 ms |
コード長 | 644 bytes |
コンパイル時間 | 196 ms |
コンパイル使用メモリ | 82,516 KB |
実行使用メモリ | 73,752 KB |
最終ジャッジ日時 | 2025-09-06 14:22:27 |
合計ジャッジ時間 | 3,364 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
def main(): n, a = map(int, input().split()) S = input() t0 = S.count("1") t1 = n - t0 #if t0 >= t1: cnt = 0 now = 0 for i in range(n): if S[i] == "0": now -= 1 elif now < 0: cnt += 1 now += 1 if cnt == a: return i+1 a -= cnt lst = [] cnt = 0 for i in range(n): if S[i] == "0": now -= 1 elif now < 0: cnt += 1 now += 1 lst.append(i+1) ans = ((a-1) // cnt + 1) * n return ans + lst[(a-1) % cnt] print(main())