結果

問題 No.36 素数が嫌い!
ユーザー frozenlibfrozenlib
提出日時 2018-06-15 13:32:21
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,577 bytes
コンパイル時間 12,451 ms
コンパイル使用メモリ 396,768 KB
実行使用メモリ 16,556 KB
最終ジャッジ日時 2024-06-30 14:47:11
合計ジャッジ時間 14,250 ms
ジャッジサーバーID
(参考情報)
judge1 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 24 WA * 2
権限があれば一括ダウンロードができます

ソースコード

diff #

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

pub fn main() {
    let i = stdin();
    let mut o = Vec::new();
    run(i.lock(), &mut o);
    stdout().write_all(&o).unwrap();
}

fn run<R: BufRead, W: Write>(i: R, o: &mut W) {
    let mut i = CpReader::new(i);
    let n = i.read::<usize>();
    let r = if solve(n) { "YES" } else { "NO" };
    writeln!(o, "{}", r).unwrap();
}
fn solve(n: usize) -> bool {
    let mut np = vec![false; (n as f64).sqrt() as usize + 1];
    for i in 2..np.len() {
        if !np[i] {
            let mut j = i * 2;
            while j < np.len() {
                np[j] = true;
                j += i;
            }
        }
    }
    let mut ps = Vec::new();
    for i in 2..np.len() {
        if !np[i] {
            ps.push(i);
        }
    }
    for &p in &ps {
        if n % p == 0 {
            let nd = n / p;
            for &p in &ps {
                if nd % p == 0 {
                    return true;
                }
            }
        }
    }
    false
}

mod utils {
    use super::*;

    pub struct CpReader<R: BufRead> {
        r: R,
        s: String,
    }
    impl<R: BufRead> CpReader<R> {
        pub fn new(r: R) -> Self {
            CpReader {
                r: r,
                s: String::new(),
            }
        }
        pub fn read_line(&mut self) -> &str {
            self.s.clear();
            self.r.read_line(&mut self.s).unwrap();
            self.s.trim()
        }

        pub fn read<T: FromStr>(&mut self) -> T {
            self.read_line().parse().ok().unwrap()
        }
    }
}
0