結果

問題 No.826 連絡網
コンテスト
ユーザー phspls
提出日時 2020-08-09 19:49:03
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 28 ms / 2,000 ms
コード長 1,578 bytes
コンパイル時間 14,344 ms
コンパイル使用メモリ 376,760 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-05 10:58:29
合計ジャッジ時間 13,089 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::HashSet;

fn get_primes(n: usize) -> Vec<bool> {
    let mut flgs = vec![true; n+1];
    flgs[0] = false;
    flgs[1] = false;
    let limit = (n as f64).sqrt().ceil() as usize;
    for i in 2..=limit {
        if !flgs[i] { continue; }
        for j in i..=n/i {
            flgs[i*j] = false;
        }
    }
    flgs
}

fn main() {
    let mut np = String::new();
    std::io::stdin().read_line(&mut np).ok();
    let np: Vec<usize> = np.trim().split_whitespace().map(|s| s.parse().unwrap()).collect();
    let n = np[0];
    let p = np[1];

    let primes = get_primes(n);
    let mut used_primes: HashSet<usize> = HashSet::new();
    if primes[p] {
        used_primes.insert(p);
    } else {
        let mut temp = p;
        for i in 2..p {
            if !primes[i] { continue; }
            while temp % i == 0 {
                temp /= i;
                used_primes.insert(i);
            }
        }
    }
    for i in 2..=n/p {
        if primes[i] {
            used_primes.insert(i);
        }
    }

    let limit: usize = if p == 9 { 0 } else { n / used_primes.iter().min().unwrap() };
    let mut result: Vec<bool> = vec![false; n+1];
    for i in 2..=n {
        if primes[i] && i <= limit {
            used_primes.insert(i);
        }
        if primes[i] && used_primes.contains(&i) {
            for j in 1..=n/i {
                result[i*j] = true;
                if primes[j] {
                    used_primes.insert(j);
                }
            }
        }
    }
    println!("{}", result.iter().filter(|&f| *f).count());
}
0