結果

問題 No.7 プライムナンバーゲーム
コンテスト
ユーザー cra77756176
提出日時 2022-11-20 00:00:01
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 132 ms / 5,000 ms
コード長 1,008 bytes
コンパイル時間 13,571 ms
コンパイル使用メモリ 377,160 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-21 04:40:03
合計ジャッジ時間 14,736 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::{collections::HashSet, io};

fn get_primes(max: usize) -> Vec<usize> {
    let mut is_prime = vec![true; max + 1];
    is_prime[0] = false;
    is_prime[1] = false;

    for n in 2..=max {
        if !is_prime[n] {
            continue;
        }
        if n * n > max {
            break;
        }
        for i in ((n * n)..=max).step_by(n) {
            is_prime[i] = false;
        }
    }

    (0..=max).filter(|&n| is_prime[n]).collect()
}

fn main() {
    let mut n = String::new();
    io::stdin().read_line(&mut n).ok();
    let n = n.trim().parse::<usize>().unwrap();

    let primes = get_primes(n);
    let mut table = vec![1; n + 1];

    for i in 2..=n {
        let mut reachable = HashSet::new();
        for &p in &primes {
            if p > i {
                break;
            }
            reachable.insert(table[i - p]);
        }
        table[i] = (0..=i).find(|n| !reachable.contains(n)).unwrap();
    }

    println!("{}", if table[n] > 0 { "Win" } else { "Lose" });
}
0