結果

問題 No.7 プライムナンバーゲーム
ユーザー tshig
提出日時 2016-08-04 15:21:49
言語 Ruby
(3.4.1)
結果
AC  
実行時間 175 ms / 5,000 ms
コード長 625 bytes
コンパイル時間 202 ms
コンパイル使用メモリ 7,424 KB
実行使用メモリ 20,480 KB
最終ジャッジ日時 2024-10-01 15:47:33
合計ジャッジ時間 2,828 ms
ジャッジサーバーID
(参考情報)
judge1 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff #

require 'prime'

class Calc0007
  def initialize(args)
    args = args.map { |l| l.chomp.split(/\s+/) }
    @n = args.shift.first.to_i
    @primes = Prime.instance.each(@n).to_a
    @memo = Array.new(@n + 1)
  end

  def calc
    dp(@n, true)
  end

  def dp(k, turn)
    if k == 0 || k == 1
      res = true
    elsif !@memo[k].nil?
      res = @memo[k]
    else
      ps = @primes.take_while { |p| p < k }.reverse
      res = ps.any? { |p| dp(k - p, !turn) == turn }
    end
    @memo[k] = res
    !(res ^ turn)
  end

  def run
    calc ? 'Win' : 'Lose'
  end
end

puts Calc0007.new(STDIN.readlines).run if __FILE__ == $0
0