結果

問題 No.649 ここでちょっとQK！
コンテスト
ユーザー letrangerjp
提出日時 2018-02-10 18:34:11
言語 Ruby
(3.4.1)
結果
AC  
実行時間 1,711 ms / 3,000 ms
コード長 1,489 bytes
コンパイル時間 125 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 18,048 KB
最終ジャッジ日時 2024-10-15 13:22:17
合計ジャッジ時間 21,761 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 32
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:79: warning: ambiguous first argument; put parentheses or a space even after `-' operator
Syntax OK

ソースコード

diff # 

class PriorityQueue
  attr_reader :elements

  def initialize
    @elements = [nil]
  end

  def <<(element)
    @elements << element
    bubble_up(@elements.size - 1)
  end

  def pop
    exchange(1, @elements.size - 1)
    max = @elements.pop
    bubble_down(1)
    max
  end

  def size
    @elements.size-1
  end

  def head
    @elements[1]
  end

  private

  def bubble_up(index)
    parent_index = (index / 2)

    return if index <= 1
    return if @elements[parent_index] >= @elements[index]

    exchange(index, parent_index)
    bubble_up(parent_index)
  end

  def bubble_down(index)
    child_index = (index * 2)

    return if child_index > @elements.size - 1

    not_the_last_element = child_index < @elements.size - 1
    left_element = @elements[child_index]
    right_element = @elements[child_index + 1]
    child_index += 1 if not_the_last_element && right_element > left_element

    return if @elements[index] >= @elements[child_index]

    exchange(index, child_index)
    bubble_down(child_index)
  end

  def exchange(source, target)
    @elements[source], @elements[target] = @elements[target], @elements[source]
  end
end

l = PriorityQueue.new
r = PriorityQueue.new

Q, K = gets.split.map &:to_i

$<.map{|s|
  if s[/^1 /]
    v = $'.to_i
    if l.size < K
      l << v
    elsif v < l.head
      r << -l.pop
      l << v
    else
      r << -v
    end
  else
    if l.size < K
      p -1
    else
      p l.pop
      l << -r.pop if r.size > 0
    end
  end
}
0