はじめに
Webエンジニアを目指して、RubyやRailsをいじってます。
今回は、RubyでAtCoder ABC257のA, B, Cを解きました。備忘録として解き方をまとめていきたいと思います。
A - A to Z String 2
a-257.rb
n, x = gets.split.map(&:to_i)
puts (65 + ((x - 1) / n)).chr
解説
N個ずつ並べるので、Aから見て(x - 1)をnで割った位置にある文字が答えとなります。
B - 1D Pawn
b-257.rb
n, k, q = gets.split.map(&:to_i)
a = gets.split.map(&:to_i)
l = gets.split.map(&:to_i)
q.times do |i|
index = l[i] - 1
next if a[index] == n
if l[i] == k
a[index] += 1
elsif a[index] + 1 < a[l[i]]
a[index] += 1
end
end
puts a.join(" ")
解説
（公式解説を参考にしました）
以下の場合分けをすることにより実装することができます。
1．Li=Kならば（K=Nの場合は除く）、K<NよりA[L[i]-1]の値をインクリメントします。
2．上記以外でかつ、次のマスにコマがなければA[L[i]-1]の値をインクリメントします。
C - Robot Takahashi
c-257.rb
# 修正前
n = gets.to_i
s = gets.chomp
w = gets.split.map(&:to_i)
array = []
w.each_with_index do |factor, index|
array << [factor, s[index]]
end
array.sort!
ans = count = s.count("1")
array.each_with_index do |factor, index|
if factor[1] == "1"
count -= 1
else
count += 1
end
if index != n - 1
ans = [ans, count].max if factor[0] != array[index + 1][0]
else
ans = [ans, count].max
end
end
puts ans
c-257.rb
# 修正後
n = gets.to_i
s = gets.chomp
w = gets.split.map(&:to_i)
array = w.zip(s.chars).sort
ans = count = s.count("1")
array.each_with_index do |(weight, child_or_adult), index|
count += (child_or_adult == "1") ? -1 : 1
- ans = [ans, count].max if array.dig(index + 1) && weight != array[index + 1][0]
+ ans = [ans, count].max if array.dig(index + 1, 0) != weight
end
puts [ans, count].max
解説
（公式解説を参考にしました）
与えられた体重を昇順にソートし、1番左から線を引き右にずらしていくことで判定していきます（線の左が子ども、右が大人）。countの初期値については、最初すべての人が大人と判定してるため大人の数が正しく判定できる人数になります。