RubyでAtCoder ABC255(A, B, C)を解いてみた

はじめに

Webエンジニアを目指して、RubyやRailsをいじってます。
今回は、RubyでAtCoder ABC255のA, B, Cを解きました。備忘録として解き方をまとめていきたいと思います。

A - You should output ARC, though this is ABC.

a-255.rb

r, c = gets.split.map(&:to_i)
a, b = gets.split.map(&:to_i)
d, e = gets.split.map(&:to_i)

if r == 1
  puts c == 1 ? a : b
else
  puts c == 1 ? d : e
end

B - Light It Up

b-255.rb

n, k = gets.split.map(&:to_i)
a = gets.split.map(&:to_i)
array = Array.new(n) { gets.split.map(&:to_i) }

ans = 0
array.each_with_index do |(x, y), index|
  next if a.include?(index + 1)
  d = a.map{ |nx| (x - array[nx - 1][0]) ** 2 + (y - array[nx - 1][1]) ** 2 }.min
  ans = [ans, d].max
end
puts Math.sqrt(ans)

解説

（公式解説を参考にしました）

その人から最も近い人が明かりをもっていれば良いので、答えはその人から最も近い人までの距離の最大値となります。

C - ±1 Operation 1

c-255.rb

def f(a, d, i)
  a + i * d
end

x, a, d, n = gets.split.map(&:to_i)

if d < 0
  a += d * (n - 1)
  d *= -1
end

if x <= a
  puts a - x
  exit
elsif x >= f(a, d, n - 1)
  puts x - f(a, d, n - 1)
  exit
end

result = (0..n - 1).bsearch { |i| f(a, d, i) >= x }
ans = [(f(a, d, result) - x).abs, (f(a, d, result - 1) - x).abs].min
puts ans

解説

（公式解説を参考にしました）

最初に、D<0の場合は与えられた等差数列を反転します。このとき答えは、Xに最も近いSの要素を二分探索して求めた値とXとの差となります。