結果

問題 No.77 レンガのピラミッド
ユーザー code-devo
提出日時 2016-01-21 09:17:45
言語 Ruby
(3.4.1)
結果
AC  
実行時間 164 ms / 5,000 ms
コード長 1,015 bytes
コンパイル時間 53 ms
コンパイル使用メモリ 7,552 KB
実行使用メモリ 12,544 KB
最終ジャッジ日時 2024-09-21 14:55:54
合計ジャッジ時間 3,751 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 5
other AC * 20
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

# うーむ・・・

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
0