結果

問題 No.2527 H and W
ユーザー neko_the_shadow
提出日時 2024-02-16 12:31:47
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 31 ms / 2,000 ms
コード長 1,274 bytes
コンパイル時間 11,682 ms
コンパイル使用メモリ 405,380 KB
実行使用メモリ 25,372 KB
最終ジャッジ日時 2024-09-28 19:17:05
合計ジャッジ時間 13,376 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 23
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::{cmp::max, io::stdin};

fn main() {
    let mut line = String::new();
    stdin().read_line(&mut line).unwrap();
    let mut tokens = line.trim().split_whitespace().map(|token| token.parse::<usize>().unwrap());
    let h = tokens.next().unwrap();
    let w = tokens.next().unwrap();
    let k = tokens.next().unwrap();

    let m = 998244353;
    let mut mncr = ModuloNCR::new(max(h, w), m);
    let mut ret = 0;
    for a in 0..=w {
        if w-a>0 && k%(w-a)==0 && h>=k/(w-a) {
            let b = h-k/(w-a);
            ret += mncr.ncr(h, b) * mncr.ncr(w, a);
            ret %= m;
        }
    }
    println!("{}", ret);
}


struct ModuloNCR {
    f: Vec<usize>,
    g: Vec<usize>,
    m: usize,
}

impl ModuloNCR {
    pub fn new(n: usize, m: usize) -> Self {
        let mut f = vec![0; n+1];
        let mut g = vec![0; n+1];
        let mut t = vec![0; n+1];
        f[0] = 1;
        f[1] = 1;
        g[0] = 1; 
        g[1] = 1;
        t[1] = 1;
        for i in 2..=n {
            f[i] = f[i-1]*i%m;
            t[i] = m - t[m%i] * (m / i) % m;
            g[i] = g[i - 1] * t[i] % m;
        }
        ModuloNCR{f, g, m}
    }

    pub fn ncr(&mut self, n: usize, r: usize) -> usize {
        self.f[n]*(self.g[r]*self.g[n-r]%self.m)%self.m
    }
}
0