結果

問題 No.7 プライムナンバーゲーム
コンテスト
ユーザー Tiramister
提出日時 2018-09-03 19:38:29
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 31 ms / 5,000 ms
コード長 1,763 bytes
コンパイル時間 12,960 ms
コンパイル使用メモリ 379,844 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-01 16:14:15
合計ジャッジ時間 12,445 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;
use std::str::FromStr;

#[allow(dead_code)]
fn get_line() -> String {
    let stdin = stdin();
    let mut line = String::new();
    stdin.lock().read_line(&mut line).expect("io error.");
    line.trim().to_string()
}

#[allow(dead_code)]
fn cast<T: FromStr>(s: &str) -> T {
    s.parse().ok().expect("parse error.")
}

#[allow(dead_code)]
fn get_vec<T: FromStr>() -> Vec<T> {
    (&get_line()).split(' ').map(cast::<T>).collect()
}


/* ---------- ここまでテンプレ ---------- */

// エラトステネスの篩で素数表を生成
fn make_primes(n: usize) -> Vec<usize> {
    let mut primes: Vec<usize> = Vec::new();

    let mut is_prime: Vec<bool> = vec![true; n+1];
    is_prime[1] = false;

    for i in 2..n+1 {
        if !is_prime[i] {continue};
        primes.push(i);

        for j in 2..n+1 {
            if i * j > n {break;}
            is_prime[i * j] = false;
        }
    }
    primes
}

// 自分の番がmで回ってきたときに勝てるならtrue
fn game(m: usize,
        dp: &mut Vec<Option<bool>>,
        primes: &Vec<usize>) -> bool
{
    if m < 2 {
        // 1手前にすでに相手が負けている
        return true;
    }

    if dp[m] != None {
        // メモを利用
        return dp[m].unwrap();
    }

    let mut ret = false;
    for &p in primes {
        if m < p {break;}

        // dp[m - p]=falseならm-pにすれば勝てる
        ret |= !game(m - p, dp, primes);
    }

    dp[m] = Some(ret);
    ret
}

fn main() {
    let n: usize = cast(&get_line());
    let primes = make_primes(n);

    let mut dp: Vec<Option<bool>> = vec![None; n + 1];
    // gameの返り値をメモ


    println!("{}",
        if game(n, &mut dp, &primes) {"Win"}
            else {"Lose"}
    );
}
0