結果

問題 No.78 クジ付きアイスバー
コンテスト
ユーザー persimmon-persimmon
提出日時 2021-06-18 11:45:16
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 972 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 187 ms
コンパイル使用メモリ 85,336 KB
実行使用メモリ 67,696 KB
最終ジャッジ日時 2026-03-06 07:47:06
合計ジャッジ時間 8,557 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 TLE * 1
other AC * 34 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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
      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