Codeforces Round #161 (Div. 2)

http://www.codeforces.com/contest/263

A, B問題だけ。

A. Beautiful Matrix

x, y = 0, 0

for i in 0 .. 4 do
  cols = gets.chomp.split(" ").map{|n| n.to_i}
  
  for j in 0 .. 4 do
    if cols[j] == 1
      x = j
      y = i
      break
    end
  end  
end
puts (2 - x).abs + (2 - y).abs

B. Squares

def solve(n, k, points)
  return [- 1] if k > n
  
  points.sort!
  
  x = points[n - k]
  y = 0
  
  return [x, y]
end


n, k = gets.chomp.split(" ").map{|n| n.to_i}
points = gets.chomp.split(" ").map{|n| n.to_i}

puts solve(n, k, points).join(" ")