結果

問題 No.78 クジ付きアイスバー
ユーザー buey_tbuey_t
提出日時 2023-04-24 13:51:58
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 162 ms / 5,000 ms
コード長 2,619 bytes
コンパイル時間 201 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 89,728 KB
最終ジャッジ日時 2024-11-08 15:53:24
合計ジャッジ時間 7,265 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

from math import sqrt,sin,cos,tan,ceil,radians,floor,gcd,exp,log,log10,log2,factorial,fsum
from heapq import heapify, heappop, heappush
from bisect import bisect_left, bisect_right
from copy import deepcopy
import copy
import random
from collections import deque,Counter,defaultdict
from itertools import permutations,combinations
from decimal import Decimal,ROUND_HALF_UP
#tmp = Decimal(mid).quantize(Decimal('0'), rounding=ROUND_HALF_UP)
from functools import lru_cache, reduce
#@lru_cache(maxsize=None)
from operator import add,sub,mul,xor,and_,or_,itemgetter
import sys
input = sys.stdin.readline
# .rstrip()
INF = 10**18
mod1 = 10**9+7
mod2 = 998244353

#DecimalならPython
#再帰ならPython!!!!!!!!!!!!!!!!!!!!!!!!!!


'''
周期性とかかな
ダブリンぐでもいけそうだけど
一周で何本もらえるのか
それで、Kをそれで割ることで、何箱使わないといけないかはわかる
そのあとは愚直でいい

いや、当たりを引き換える必要があるから
周期性が乱れる
一週目だけ特別なのか

当たりくじの数を記録しておいて、当たりくじがアイスバーの長さになるたびにやる
それで結構いい近似を得れる
そして、while now < K+N+atari でやる

Nが小さいと全然高速化になってないんだよな
二分探索とか?
何本買うかで二分探索すると
思いつく方法は、まあ買う数さえ決まればあたりの数は容易に計算できる
あたりの数によって変わるものを動的に処理したい
どこから始まるかも大事だし...
'''

N,K = map(int, input().split())
S = list(input().rstrip())

atari = 0
for i in range(N):
    if S[i] == '1':
        atari += 1
    elif S[i] == '2':
        atari += 2

if atari >= N:
    cst = 0
    rem = 0
    i = 0
    while i < N and cst < K:
        if rem > 0:
            rem -= 1
        else:
            cst += 1
        
        if S[i] == '1':
            rem += 1
        elif S[i] == '2':
            rem += 2
        
        i += 1
    
    print(cst)
    exit()

l = -1
h = K+1

while h-l > 1:
    m = (h+l)//2
    
    cnt = m-m%N
    rem = m%N + atari*(m//N)
    
    while atari*(rem//N) > 0:
        cnt += rem-rem%N
        rem = atari*(rem//N) + rem%N
        
    
    i = 0
    while rem > 0 and cnt < K:
        if S[i] == '1':
            rem += 1
        elif S[i] == '2':
            rem += 2
        
        rem -= 1
        cnt += 1
        
        i += 1
        
        if i == N:
            i = 0
    
    if cnt >= K:
        h = m
    else:
        l = m

print(h)
0