結果

問題 No.407 鴨等素数間隔列の数え上げ
ユーザー aimyaimy
提出日時 2017-10-24 10:22:40
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 50 ms / 1,000 ms
コード長 2,833 bytes
コンパイル時間 12,528 ms
コンパイル使用メモリ 383,964 KB
実行使用メモリ 9,512 KB
最終ジャッジ日時 2024-11-21 17:35:00
合計ジャッジ時間 14,488 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

use prime::*;

fn main() {
  let (n, l): (usize, usize) = get::tuple();

  let d = l / (n-1);
  let ps = primes(d);
  let ans = ps.iter().map(|p|l-p*(n-1)+1).sum::<usize>();
  
  println!("{}", ans)
}

#[allow(dead_code)]
mod prime {
  pub fn is_prime(n: usize) -> bool {
    let sq = (n as f64).sqrt().floor() as usize;
    let p = (2..sq+1).all(|d| n%d != 0);
    p
  }

  pub fn primes(n: usize) -> Vec<usize> {
    if n < 2 {return vec![]}
    let mut ps = vec![true; n+1];
    ps[0] = false;
    ps[1] = false;
    for i in 2 .. n+1 {
      if ps[i] {
        for k in 2 .. n/i+1 {
          ps[i*k] = false;
        }
      }
    }
    ps.iter().enumerate().filter(|&(_,&p)|p).map(|(i,_)|i).collect()
  }

  pub fn factorize(n: usize) -> Vec<usize> {
    let sq = (n as f64).sqrt().floor() as usize;
    let mut fs = vec![];
    let mut x = n;
    for d in 2 .. sq+1 {
      if x == 1  {break}
      while x > 1 && x % d == 0 {
        fs.push(d);
        x /= d;
      }
    }
    if x != 1 {fs.push(x)};
    fs
  }
}

#[allow(dead_code)]
mod get {
  use std::io::*;
  use std::str::*;

  pub fn val<T: FromStr>() -> T {
    let mut buf = String::new();
    stdin().read_line(&mut buf).ok();
    buf.trim().parse::<T>().ok().unwrap()
  }

  pub fn vals<T: FromStr>(n: usize) -> Vec<T> {
    let mut vec: Vec<T> = vec![];
    for _ in 0 .. n {
      vec.push(val());
    }
    vec
  }

  pub fn tuple<T1: FromStr, T2: FromStr>() -> (T1, T2) {
    let mut buf = String::new();
    stdin().read_line(&mut buf).ok();
    let mut it = buf.trim().split_whitespace();
    let x = it.next().unwrap().parse::<T1>().ok().unwrap();
    let y = it.next().unwrap().parse::<T2>().ok().unwrap();
    (x, y)
  }

  pub fn tuples<T1: FromStr, T2: FromStr>(n: usize) -> Vec<(T1, T2)> {
    let mut vec: Vec<(T1, T2)> = vec![];
    for _ in 0 .. n {
      vec.push(tuple());
    }
    vec
  }

  pub fn tuple3<T1: FromStr, T2: FromStr, T3: FromStr>() -> (T1, T2, T3) {
    let mut buf = String::new();
    stdin().read_line(&mut buf).ok();
    let mut it = buf.trim().split_whitespace();
    let x = it.next().unwrap().parse::<T1>().ok().unwrap();
    let y = it.next().unwrap().parse::<T2>().ok().unwrap();
    let z = it.next().unwrap().parse::<T3>().ok().unwrap();
    (x, y, z)
  }

  pub fn tuple3s<T1: FromStr, T2: FromStr, T3: FromStr>(n: usize) -> Vec<(T1, T2, T3)> {
    let mut vec: Vec<(T1, T2, T3)> = vec![];
    for _ in 0 .. n {
      vec.push(tuple3());
    }
    vec
  }

  pub fn list<T: FromStr>() -> Vec<T> {
    let mut buf = String::new();
    stdin().read_line(&mut buf).ok();
    buf.trim().split_whitespace().map(|t| t.parse::<T>().ok().unwrap()).collect()
  }

  pub fn lists<T: FromStr>(h: usize) -> Vec<Vec<T>> {
    let mut mat: Vec<Vec<T>> = vec![];
    for _ in 0 .. h {
      mat.push(list());
    }
    mat
  }
}
0