Submission #402378

---

Source Code Expand

R,C,K = gets.split.map(&:to_i)
N = gets.to_i
drops = N.times.map{gets.split.map(&:to_i)}

row_count = drops.group_by{|r,c| r}.to_h
count_row = row_count.inject(Hash.new{|h,k| h[k] = []}){|s,h| s[h.last.size] << h.first; s}
col_count = drops.group_by{|r,c| c}.to_h
count_col = col_count.inject(Hash.new{|h,k| h[k] = []}){|s,h| s[h.last.size] << h.first; s}
count = 0

row_count.each do |n,ra|
  count += ra.size * count_col[K - n].size
end

drops.each do |r,c|
  case row_count[r] + col_count[c]
  when K; count -= 1
  when K + 1; count += 1
  end
end

puts count

Submission Info

Compile Error

./Main.rb:6: warning: assigned but unused variable - count_row

Judge Result