はじめに
Webエンジニアを目指して、RubyやRailsをいじってます。
今回は、RubyでAtCoder ABC268のA, B, Cを解きました。備忘録として解き方をまとめていきたいと思います。
A - Five Integers
a-268.rb
puts gets.split.uniq.size
解説
uniqメソッドで重複を取り除いた後の大きさを出力すればOKです。
B - Prefix?
b-268.rb
# 修正前
s = gets.chomp.chars
t = gets.chomp.chars
s.each_with_index do |factor, index|
if factor != t[index]
puts "No"
exit
end
end
puts "Yes"
b-268.rb
# 修正後
s = gets.chomp
t = gets.chomp[0, s.size]
puts s == t ? "Yes" : "No"
別解
s = gets.chomp
t = gets.chomp
puts t.start_with?(s) ? "Yes" : "No"
解説
先頭から順にsとtが一致しているか調べていき、途中で一致しないものがあればNoを出力して終了、すべて一致していればYesを出力します。
C - Chinese Restaurant
c-268.rb
n = gets.to_i
m = gets.split.map(&:to_i)
count = Array.new(n, 0)
n.times do |i|
j = m[i] - i
count[(j - 1) % n] += 1
count[j % n] += 1
count[(j + 1) % n] += 1
end
puts count.max
解説
（公式解説を参考にしました）
演算数を減らしてより高速にしています。