結果

問題 No.889 素数!
コンテスト
ユーザー wesrivery
提出日時 2019-09-21 17:44:01
言語 Rust
(1.83.0 + proconio)
結果
RE  
実行時間 -
コード長 2,714 bytes
コンパイル時間 14,873 ms
コンパイル使用メモリ 378,160 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-19 03:00:59
合計ジャッジ時間 16,583 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 45 WA * 15 RE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::HashMap;

#[allow(unused_macros)]
macro_rules! input {
    ( $($t:ty),* ) => {{
        let mut s = String::new();
        std::io::stdin().read_line(&mut s);
        let mut splits = s.trim().split_whitespace();
        ($( { splits.next().unwrap().parse::<$t>().unwrap() },)*)
    }}
}

#[allow(dead_code)]
fn primes(n: u64) -> Vec<u64> {
    if n < 3 {
        return vec![];
    }
    let mut res: Vec<u64> = vec![2];
    for i in 3..n {
        let i = i as u64;
        let root = ((n as f64).sqrt().ceil() as usize + 1) as u64;
        let composite = res.iter().take_while(|&&x| x <= root).any(|x| i % x == 0);
        if !composite {
            res.push(i);
        }
    }
    return res;
}

fn prime_factorization(n: u64) -> std::collections::HashMap<u64, u32> {
    let primes = primes((n / 2) + 2);
    let mut h = std::collections::HashMap::new();

    let mut current = n;
    while current != 1 {
        let mut found = false;
        for p in &primes {
            if current % p == 0 {
                let count = h.entry(*p).or_insert(0);
                *count += 1;
                current /= p;
                found = true;
            }
        }
        if !found {
            let count = h.entry(n).or_insert(0);
            *count += 1;
            break;
        }
    }

    return h;
}

fn count_values(v: Vec<u32>) -> HashMap<u32, u32> {
    let mut h = HashMap::new();
    
    for val in v {
        let count = h.entry(val).or_insert(0);
        *count += 1;
    }

    h
}


#[allow(dead_code)]
fn divisors(n: u64) -> Vec<u64> {
    let mut res = vec![];
    for i in (1..(((n as f64).sqrt().floor() as u64) + 1)).rev() {
        if n % i == 0 {
            res.insert(0, i);
            if i != n / i {
                res.push(n / i);
            }
        }
    }
    return res;
}

#[allow(unused_must_use)]
#[allow(unused_variables)]
fn solve() {
    let (n,) = input!(u64);

    if n == 1 {
        println!("1");
        return;
    }

    let h = prime_factorization(n);

    let vals: Vec<u32> = h.values().map(|&a| a).collect();
    let same_value_count = count_values(vals);

    let divs = divisors(n);
    let divisor_total: &u64 = &divs[..(divs.len() - 1)].iter().sum();

    if same_value_count.keys().next().unwrap() == &1 && same_value_count.values().next().unwrap() == &1 {
        println!("Sosu!");
    } else if same_value_count.keys().next().unwrap() % &2 == 0 {
        println!("Heihosu!");
    } else if same_value_count.keys().next().unwrap() % &3 == 0 {
        println!("Ripposu!");
    } else if divisor_total == &n {
        println!("Kanzensu!");
    } else {
        println!("{}", n);
    }
}

fn main() {
    solve();
}
0