AtCoder ARC #041 B - アメーバ

arc041-b-editorial.rb

# Editorial ver.

N, M = gets.chomp.split(" ").map(&:to_i)

@m = []
N.times do
  @m << gets.chomp.split("").map(&:to_i)
end
@ans = Array.new(N){Array.new(M, 0)}
(N-2).times do |i|
  (1...(M-1)).each do |j|
    x = @m[i][j]
    @ans[i+1][j] += x
    @m[i][j] -= x
    @m[i+1][j-1] -= x
    @m[i+1][j+1] -= x
    @m[i+2][j] -= x
  end
end
@ans.each do |row|
  puts row.join("")
end

arc041-b.rb

N, M = gets.chomp.split(" ").map(&:to_i)

@m = []
N.times do
  @m << gets.chomp.split("").map(&:to_i)
end
@ans = Array.new(N){Array.new(M, 0)}
(1...(N-1)).each do |i|
  (1...(M-1)).each do |j|
    x = [@m[i-1][j], @m[i][j+1], @m[i+1][j], @m[i][j-1]].min
    @ans[i][j] = x
    @m[i-1][j] -= x
    @m[i][j+1] -= x
    @m[i+1][j] -= x
    @m[i][j-1] -= x
  end
end
@ans.each do |row|
  puts row.join("")
end