結果

問題 No.3656 Game Scores and Costs
コンテスト
ユーザー tomerun
提出日時 2026-08-30 13:35:59
言語 Crystal
(1.21.0)
コンパイル:
crystal build -Donline_judge -o a.out --release --no-debug _filename_
実行:
./a.out
結果
AC  
実行時間 36 ms / 2,000 ms
+ 826µs
コード長 1,607 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 9,552 ms
コンパイル使用メモリ 339,104 KB
実行使用メモリ 19,712 KB
最終ジャッジ日時 2026-08-30 13:37:08
合計ジャッジ時間 12,072 ms
ジャッジサーバーID
（参考情報） 		judge3_0 / judge2_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 21
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

n, k, x = read_line.split.map(&.to_i)
a = read_line.split.map(&.to_i64)
q = PriorityQueue(Int64).new(k)
ans = -x
sum = 0i64
k.times do |i|
  q.add(-a[i])
  sum += a[i]
  ans = {ans, sum - x.to_i64 * (i + 1)}.max
end
k.upto(n - 1) do |i|
  if a[i] > -q.top
    sum += q.top
    q.pop
    q.add(-a[i])
    sum += a[i]
  end
  ans = {ans, sum - x.to_i64 * (i + 1)}.max
end
puts ans

class PriorityQueue(T)
  def initialize(capacity : Int32)
    @elem = Array(T).new(capacity)
  end

  def initialize(list : Enumerable(T))
    @elem = list.to_a
    1.upto(size - 1) { |i| fixup(i) }
  end

  def size
    @elem.size
  end

  def add(v)
    @elem << v
    fixup(size - 1)
  end

  def top
    @elem[0]
  end

  def pop
    ret = @elem[0]
    last = @elem.pop
    if size > 0
      @elem[0] = last
      fixdown(0)
    end
    ret
  end

  def clear
    @elem.clear
  end

  def decrease_top(new_value : T)
    @elem[0] = new_value
    fixdown(0)
  end

  def to_s(io : IO)
    io << @elem
  end

  private def fixup(index : Int32)
    while index > 0
      parent = (index - 1) // 2
      break if @elem[parent] >= @elem[index]
      @elem[parent], @elem[index] = @elem[index], @elem[parent]
      index = parent
    end
  end

  private def fixdown(index : Int32)
    while true
      left = index * 2 + 1
      break if left >= size
      right = index * 2 + 2
      child = right >= size || @elem[left] > @elem[right] ? left : right
      if @elem[child] > @elem[index]
        @elem[child], @elem[index] = @elem[index], @elem[child]
        index = child
      else
        break
      end
    end
  end
end
0