結果

問題 No.660 家を通り過ぎないランダムウォーク問題
ユーザー pekempeypekempey
提出日時 2018-03-02 23:04:55
言語 Rust
(1.83.0 + proconio)
結果
WA  
実行時間 -
コード長 1,303 bytes
コンパイル時間 13,317 ms
コンパイル使用メモリ 397,864 KB
実行使用メモリ 25,700 KB
最終ジャッジ日時 2024-06-22 05:57:06
合計ジャッジ時間 16,434 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 21 WA * 24
権限があれば一括ダウンロードができます

ソースコード

diff #

const MOD: i64 = 1000000007;

fn main() {
  let n: usize = input();  

  let mut inv: Vec<i64> = vec![0; 1010101];
  let mut fact: Vec<i64> = vec![0; 1010101];
  let mut ifact: Vec<i64> = vec![0; 1010101];

  inv[1] = 1;
  fact[0] = 1;
  fact[1] = 1;
  ifact[0] = 1;
  ifact[1] = 1;

  for i in 2..inv.len() {
    inv[i] = inv[(MOD % (i as i64)) as usize] * (MOD - MOD / (i as i64)) % MOD;
    fact[i] = (i as i64) * fact[i - 1] % MOD;
    ifact[i] = inv[i] * ifact[i - 1] % MOD;
  }

  let f = |x: usize, y: usize| {
    if (x + y) % 2 == 1 {
      0
    } else {
      let n = x;
      let r = (x + y) / 2;
      if n >= r {
        fact[n] * ifact[r] % MOD * ifact[n - r] % MOD
      } else {
        0
      }
    }
  };

  let mut ans: i64 = 0;
  for i in 1..(2*n + 1) {
    ans += f(i - 1, n - 1) - f(i - 1, n + 1);
  }

  println!("{}", ans);
}

fn input<T: std::str::FromStr>() -> T {
  use std::io::Read;
  let stdin = std::io::stdin();
  let stdin = stdin.bytes();
  let token = stdin
    .skip_while(|x| (*x.as_ref().unwrap() as char).is_whitespace())
    .take_while(|x| !(*x.as_ref().unwrap() as char).is_whitespace())
    .map(|x| x.unwrap())
    .collect();
  let token = String::from_utf8(token).unwrap();
  match token.parse() {
    Ok(x) => x,
    Err(_) => panic!("{}", token),
  }
}
0