結果

問題 No.198 キャンディー・ボックス２
ユーザー siman
提出日時 2016-03-27 02:17:49
言語 Ruby
(3.4.1)
結果
AC  
実行時間 86 ms / 1,000 ms
コード長 1,166 bytes
コンパイル時間 56 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 12,288 KB
最終ジャッジ日時 2024-11-16 09:55:38
合計ジャッジ時間 3,397 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:13: warning: assigned but unused variable - answer
Syntax OK

ソースコード

diff # 

class Yukicoder
  attr_reader :candy, :remain

  def initialize
    @remain = gets.to_i
    n = gets.to_i
    @candy = []

    n.times do
      @candy << gets.to_i
    end

    answer = 0

    if candy.size == 1
      puts "0"
      return
    end

    a = 0
    b = candy.max
    answer = candy.inject(:+)

    while a < b-2
      c = a + (b-a)/3
      d = [a + 2 * ((b-a)/3), b].min

      v1 = calc(c)
      v2 = calc(d)

      if v1 < v2 || v2 == Float::INFINITY
        b = d
      else
        a = c+1
      end
    end

    min_val = Float::INFINITY

    (a-100).upto(b+100) do |i|
      up_val = 0
      down_val = 0

      candy.each do |c|
        if c < i
          up_val += i-c
        else
          down_val += c-i
        end
      end

      if up_val <= remain + down_val
        min_val = [min_val, down_val + up_val].min
      end
    end

    puts min_val
  end

  def calc(i)
    up_val = 0
    down_val = 0

    candy.each do |c|
      if c < i
        up_val += i-c
      else
        down_val += c-i
      end
    end

    if up_val <= remain + down_val
      down_val + up_val
    else
      Float::INFINITY
    end
  end
end

Yukicoder.new
0