結果

問題 No.78 クジ付きアイスバー
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-06-18 11:46:40
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 39 ms / 5,000 ms
コード長 1,001 bytes
コンパイル時間 264 ms
コンパイル使用メモリ 82,536 KB
実行使用メモリ 53,684 KB
最終ジャッジ日時 2024-06-22 02:16:59
合計ジャッジ時間 2,982 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

def main0(n,k,s):
  cnt=0
  ans=0
  atari=0
  while cnt<k:
    for x in s:
      if atari:
        atari-=1
      else:
        ans+=1
      atari+=x
      cnt+=1
      if cnt==k:break
  return ans

def main1(n,k,s):
  cnt=0
  ans=0
  atari=0
  for x in s:
    if atari:
      atari-=1
    else:
      ans+=1
    atari+=x
    cnt+=1
    if cnt==k:return ans
  if atari>=ans:return ans
  d={}
  cnt=0
  ans=0
  atari=0
  while True:
    for i in range(n):
      x=s[i]
      if atari:
        atari-=1
      else:
        ans+=1
      atari+=x
      cnt+=1
      if cnt==k:return ans
      if (i,atari) in d:
        cnt0,ans0=d[(i,atari)]
        dcnt=cnt-cnt0
        dans=ans-ans0
        loopcnt=(k-cnt)//dcnt
        ans+=loopcnt*dans
        cnt+=loopcnt*dcnt
        if cnt==k:return ans
      else:
        d[(i,atari)]=cnt,ans

if __name__=='__main__':
  n,k=map(int,input().split())
  s=list(map(int,list(input())))
  #ret0=main0(n,k,s)
  ret1=main1(n,k,s)
  print(ret1)
  #print(ret0,ret1)
0