結果

問題 No.78 クジ付きアイスバー
ユーザー code-devo
提出日時 2016-01-22 03:15:19
言語 Ruby
(3.4.1)
結果
AC  
実行時間 92 ms / 5,000 ms
コード長 874 bytes
コンパイル時間 609 ms
コンパイル使用メモリ 7,296 KB
実行使用メモリ 12,288 KB
最終ジャッジ日時 2024-11-08 01:36:14
合計ジャッジ時間 5,187 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

# なぜ自分のコードはこんなに長いのか(´・ω・`)

N, K = gets.split.map(&:to_i)
S = gets.chomp

def calc(want, kuji)
  raise unless want > 0

  buy = 0
  for i in 0...N do
    if kuji > 0 then
      kuji -= 1
    else
      buy += 1
    end
    kuji += S[i].to_i
    want -= 1
    break if want <= 0
  end
  return buy, kuji
end

buy_sum = 0
want = K

buy_prev = nil
kuji_prev = 0
while want > 0 do
  buy, kuji = calc(want, kuji_prev)
  break if buy <= 0 && kuji >= kuji_prev #クジだけでやっていける
  if buy == buy_prev && kuji == kuji_prev then #パターンに入った
    q, r = want.divmod(N)
    buy_sum += buy * q
    want = r

    if want > 0 then
      buy, kuji = calc(want, kuji_prev)
      buy_sum += buy
      want -= N
    end
  else
    buy_sum += buy
    want -= N
  end

  buy_prev = buy
  kuji_prev = kuji
end

puts buy_sum
0