結果

問題 No.3296 81-like number
コンテスト
ユーザー northward
提出日時 2025-12-09 20:36:25
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,633 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 12,116 ms
コンパイル使用メモリ 401,496 KB
実行使用メモリ 7,848 KB
最終ジャッジ日時 2025-12-09 20:36:39
合計ジャッジ時間 13,439 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#![allow(non_snake_case, unused_must_use, unused_imports)]
use std::io::{self, prelude::*};

fn main() {
    let (stdin, stdout) = (io::read_to_string(io::stdin()).unwrap(), io::stdout());
    let (mut stdin, mut buffer) = (stdin.split_whitespace(), io::BufWriter::new(stdout.lock()));

    macro_rules! input {
        ($t: tt, $n: expr) => {
            (0..$n).map(|_| input!($t)).collect::<Vec<_>>()
        };
        (Chars) => {
            input!(String).chars().collect::<Vec<_>>()
        };
        (Usize1) => {
            stdin.next().unwrap().parse::<usize>().unwrap() - 1
        };
        ($t: ty) => {
            stdin.next().unwrap().parse::<$t>().unwrap()
        };
    }

    let N = input!(usize);

    let mut ans = 0;
    const IS_PRIME: [bool; 100001] = is_prime();

    for p in 2.. {
        if p * p > N {
            break;
        }

        if IS_PRIME[p] {
            let mut q = p * p;

            while q <= N {
                if q * p > N {
                    ans += q;
                    break;
                }

                ans += q + q * p;
                q *= p * p;
            }
        }
    }

    writeln!(buffer, "{}", ans);
}

const fn is_prime() -> [bool; 100_001] {
    let mut a = [true; 100_001];

    a[0] = false;
    a[1] = false;

    let mut p = 1;

    loop {
        p += 1;

        if p == 100_001 {
            break;
        }

        if !a[p] {
            continue;
        }

        let mut q = p;

        loop {
            q += p;
            if q >= 100_001 {
                break;
            }

            a[q] = false;
        }
    }

    a
}
0