結果

問題 No.115 遠足のおやつ
ユーザー code-devo
提出日時 2016-01-20 16:39:25
言語 Ruby
(3.4.1)
結果
AC  
実行時間 98 ms / 5,000 ms
コード長 589 bytes
コンパイル時間 323 ms
コンパイル使用メモリ 8,320 KB
実行使用メモリ 13,056 KB
最終ジャッジ日時 2025-01-03 01:10:34
合計ジャッジ時間 5,001 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 40
権限があれば一括ダウンロードができます
コンパイルメッセージ
Main.rb:31: warning: ambiguous first argument; put parentheses or a space even after `-` operator
Syntax OK

ソースコード

diff #

class Array
  def sum
    self.size > 0 ? self.inject(&:+) : 0
  end
end

N, D, K = gets.split.map(&:to_i)

ns = *(1..N)
a = ns.shift(K)
b = []

# bになるべく大きい数値を詰めることで、aには小さい数値が残る。
while a.size > 0 && ns.size > 0 && a.sum + b.sum - a[-1] + ns[-1] < D do
  ns.unshift a.pop
  n = ns.pop
  b.unshift n
end

if a.sum + b.sum < D then
  idx = D - (a.sum + b.sum) - 1
  if ns[idx] then
    b.unshift ns.delete_at(idx)
    a.pop
  end
end

if a.size + b.size == K && a.sum + b.sum == D then
  puts a.concat(b).join(" ")
else
  puts -1
end
0