結果

問題 No.732 3PrimeCounting
ユーザー 特命ログイン
提出日時 2018-09-07 23:12:57
言語 Rust
(1.83.0 + proconio)
結果
MLE  
実行時間 -
コード長 1,769 bytes
コンパイル時間 13,381 ms
コンパイル使用メモリ 402,900 KB
実行使用メモリ 524,724 KB
最終ジャッジ日時 2024-11-29 19:43:28
合計ジャッジ時間 97,532 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
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 as i64);
        }
    }
    for i in 0..tb.len() {
        tb[i].sort();
    }
    qs.reverse();
    let mut ans = 0_i64;
    for &p in ps.iter() {
        let p = p as i64;
        for &q in qs.iter() {
            let q = q as i64;
            let r = p - q;
            if r < 0 { continue; }
            if r > (q - 2) + (q - 1) { break; }
            let r = r as usize;
            if tb[r].len() == 0 { continue; }
            let mut lp = 0;
            let mut rp = tb[r].len();
            while lp < rp {
                let hp = (rp + lp) / 2;
                if tb[r][hp] >= q {
                    rp = hp;
                } else {
                    lp = hp + 1;
                }
            }
            // println!("{} {} {} {:?}", p, q, r, tb[r]);
            ans += rp as i64;
        }
    }
    // println!("{:?}", tb);
    println!("{}", ans);
}
0