結果

問題 No.59 鉄道の旅
ユーザー tshig
提出日時 2016-08-19 10:42:43
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,344 ms / 5,000 ms
コード長 802 bytes
コンパイル時間 399 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 29,312 KB
最終ジャッジ日時 2024-12-24 22:39:06
合計ジャッジ時間 5,633 ms
ジャッジサーバーID
(参考情報)
judge3 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 12
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

class Calc0059
  def initialize(args)
    args = args.map { |l| l.chomp.split(/\s+/) }
    @n, @k = args.shift.map(&:to_i)
    @ws = args.map(&:first).map(&:to_i)
  end

  def run
    stock = []
    @ws.each do |w|
      if w > 0
        if stock.size < @k || stock[-@k] < w
          add(stock, w)
        end
      else
        remove(stock, -w)
      end
    end
    stock.size
  end

  def add(stock, w)
    i = stock.bsearch_index { |x| x >= w }
    if i
      stock.insert(i, w)
    else
      stock.push(w)
    end
  end

  def remove(stock, w)
    i = stock.bsearch_index { |x| x >= w }
    if i
      if stock[i] == w
        stock.delete_at(i)
      end
    else
      if stock[0] == w
        stock.shift
      end
    end
  end
end

puts Calc0059.new(STDIN.readlines).run if __FILE__ == $0
0