結果

問題 No.929 よくあるボールを移動するやつ
ユーザー siman
提出日時 2020-09-16 09:31:16
言語 Ruby
(3.4.1)
結果
WA  
実行時間 -
コード長 491 bytes
コンパイル時間 55 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 21,504 KB
最終ジャッジ日時 2024-06-22 03:04:30
合計ジャッジ時間 2,410 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3 WA * 1
other AC * 6 WA * 6
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

N = gets.to_i
B = gets.split.map(&:to_i)

stock = []
empty = []
ans = 0

B.each_with_index do |b, idx|
  if b == 0
    if stock.empty?
      empty << idx
    else
      i, cnt = stock.pop

      ans += idx - i
      cnt -= 1

      stock.push([i, cnt]) if cnt > 0
    end
  else
    while empty.size > 0 && b > 0
      i = empty.pop
      ans += idx - i
      b -= 1
    end

    r = b - 1

    if r <= 0
      empty.push(idx)
    else
      stock.push([idx, r])
    end
  end
end

puts ans
0