# うーむ・・・ N = gets.to_i A = gets.split.map(&:to_i) SUM = A.inject(&:+) min_move = nil for r in 1..100 do #ピラミッドの段数 py = r ** 2 #ピラミッドの石数 break if py > SUM #石が足りないなら終了 for m in (r-1)..(N-1) do #ピラミッドの中央の列 pl = 0 mi = 0 (m-r+1).upto(m+r-1).with_index do |c, idx| s = r - (r - idx - 1).abs #c列に必要な石数 d = (c < N ? A[c] : 0) - s pl += d if d > 0 mi += -d if d < 0 end sum = SUM #存在する石数 move1 = [pl, mi].min #ピラミッド構築範囲内で移動する。 move2 = (pl - mi).abs #ピラミッド構築範囲内で余った石は捨てる。足りないなら構築範囲外から持ってくる。 sum -= move2 if pl - mi >= 0 #捨てた場合、石が減る。 move3 = sum - py #ピラミッド構築範囲外の石を捨てる。 move = move1 + move2 + move3 min_move = move if !min_move || move < min_move end end puts min_move