# @param {Integer[]} digits
# @return {Integer[]}
def plus_one(digits)
  (digits.join.to_i + 1).to_s.split("").map(&:to_i)
end

# p plus_one [4, 3, 2, 1]