結果

問題 No.732 3PrimeCounting
ユーザー 特命ログイン
提出日時 2018-09-07 23:18:32
言語 Rust
(1.83.0 + proconio)
結果
MLE  
実行時間 -
コード長 1,420 bytes
コンパイル時間 16,502 ms
コンパイル使用メモリ 380,696 KB
実行使用メモリ 524,584 KB
最終ジャッジ日時 2024-11-29 20:34:17
合計ジャッジ時間 100,566 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 70 TLE * 17 MLE * 2
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::{Read, stdin};

fn main() {
    let mut buf = String::new();
    stdin().read_to_string(&mut buf).unwrap();
    let mut tok = buf.split_whitespace();
    let mut get = || tok.next().unwrap();
    macro_rules! get {
        ($t:ty) => (get().parse::<$t>().unwrap());
        () => (get!(usize));
    }
    
    let n = get!();
    let mut qs = vec![];
    let mut ps = vec![];
    let mut flag = vec![false; n * 3 + 1];
    for i in 2..flag.len() {
        if flag[i] { continue; }
        if i <= n {
            qs.push(i);
        }
        ps.push(i);
        for j in 2..flag.len() {
            let k = i * j;
            if k >= flag.len() { break; }
            flag[k] = true;
        }
    }
    let mut tb = vec![vec![]; n * 2 + 1];
    for i in 1..qs.len() {
        let q1 = qs[i];
        for j in i+1..qs.len() {
            let q2 = qs[j];
            tb[q1+q2].push(q2);
        }
    }
    let mut ans = 0_i64;
    for i in 3..tb.len() {
        tb[i].sort();
        for &q in qs.iter() {
            if flag[q+i] { continue; }
            let mut lp = 0;
            let mut rp = tb[i].len();
            while lp < rp {
                let hp = (lp + rp) / 2;
                if tb[i][hp] >= q {
                    rp = hp;
                } else {
                    lp = hp + 1;
                }
            }
            ans += rp as i64;
        }
    }
    println!("{}", ans);
}
0