結果

問題 No.1708 Quality of Contest
ユーザー 小野寺健小野寺健
提出日時 2021-10-30 15:55:30
言語 Ruby
(3.3.0)
結果
TLE  
実行時間 -
コード長 2,536 bytes
コンパイル時間 73 ms
コンパイル使用メモリ 7,680 KB
実行使用メモリ 20,380 KB
最終ジャッジ日時 2024-04-16 16:23:33
合計ジャッジ時間 5,501 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 88 ms
17,664 KB
testcase_01 AC 88 ms
12,288 KB
testcase_02 AC 96 ms
12,160 KB
testcase_03 AC 145 ms
13,184 KB
testcase_04 AC 147 ms
13,184 KB
testcase_05 AC 159 ms
12,672 KB
testcase_06 AC 135 ms
13,184 KB
testcase_07 AC 135 ms
13,184 KB
testcase_08 AC 87 ms
12,288 KB
testcase_09 TLE -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:30: warning: mismatched indentations at 'end' with 'else' at 28
Main.rb:119: warning: assigned but unused variable - k
Syntax OK

ソースコード

diff # 

class PriorityQueue
  def initialize(array = [])
    @data = []
    array.each{|a| push(a)}
  end

  def push(element)
    @data.push(element)
    bottom_up
  end

  def pop
    if size == 0
      return nil
    elsif size == 1
      return @data.pop
    else
      min = @data[0]
      @data[0] = @data.pop
      top_down
      return min
    end
  end
  
  def poll
    if size == 0
      return nil
    else
      return @data[0]
	end
  end
  
  def size
    @data.size
  end
  
  def data
	@data
  end

  private

  def swap(i, j)
    @data[i], @data[j] = @data[j], @data[i]
  end

  def parent_idx(target_idx)
    (target_idx - (target_idx.even? ? 2 : 1)) / 2
  end

  def bottom_up
    target_idx = size - 1
    return if target_idx == 0
    parent_idx = parent_idx(target_idx)
    while (@data[parent_idx][0] < @data[target_idx][0])
      swap(parent_idx, target_idx)
      target_idx = parent_idx
      break if target_idx == 0
      parent_idx = parent_idx(target_idx)
    end
  end

  def top_down
    target_idx = 0

    # child がある場合
    while (has_child?(target_idx))

      a = left_child_idx(target_idx)
      b = right_child_idx(target_idx)

      if @data[b].nil?
        c = a
      else
        c = @data[a][0] >= @data[b][0] ? a : b
      end

      if @data[target_idx][0] < @data[c][0]
        swap(target_idx, c)
        target_idx = c
      else
        return
      end
    end
  end

  # @param Integer
  # @return Integer
  def left_child_idx(idx)
    (idx * 2) + 1
  end

  # @param Integer
  # @return Integer
  def right_child_idx(idx)
    (idx * 2) + 2
  end

  # @param Integer
  # @return Boolent
  def has_child?(idx)
    ((idx * 2) + 1) < @data.size
  end
end

N, M, X = gets.split(" ").map{|s| s.to_i}

A = Array.new(M) {Array.new}

N.times {
	a, b = gets.split(" ").map{|s| s.to_i}
	A[b-1] << a
}

A.each {|a|
	a.sort_by!{|x| -x}
}

k = gets.to_i
c = gets.split(" ").map{|s| s.to_i}

cnt = []
cpos = Array.new(M, 0)
same = PriorityQueue.new
diff = PriorityQueue.new
0.upto(M-1) {|j|
	diff.push([A[j][0], j]) if A[j].length > 0
}

c.max.times {
	sa, sb = same.size > 0 ? same.poll : [-Float::INFINITY, -1]
	da, db = diff.size > 0 ? diff.poll : [-Float::INFINITY, -1]

	if sa > da + X then
		same.pop
		cpos[sb] += 1
		if cpos[sb] < A[sb].length then
			same.push([A[sb][cpos[sb]], sb])
		end
		cnt << sa
	else
		diff.pop
		cpos[db] += 1
		if cpos[db] < A[db].length 
			same.push([A[db][cpos[db]], db])
		end
		cnt << da + X
	end
}

total = 0
c.each {|i|
	total += cnt[0,i].sum
}
puts total
0