結果

問題 No.1688 Veterinarian
コンテスト
ユーザー emak
提出日時 2021-09-24 22:37:07
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 424 ms / 3,000 ms
コード長 4,014 bytes
コンパイル時間 13,719 ms
コンパイル使用メモリ 391,484 KB
実行使用メモリ 164,736 KB
最終ジャッジ日時 2024-07-05 11:01:00
合計ジャッジ時間 17,638 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます
コンパイルメッセージ
warning: unused variable: `m`
  --> src/main.rs:20:9
   |
20 |     let m = 51;
   |         ^ help: if this is intentional, prefix it with an underscore: `_m`
   |
   = note: `#[warn(unused_variables)]` on by default

ソースコード

diff #
raw source code 

#![allow(unused_macros, unused_imports)]
macro_rules! dbg {
    ($($xs:expr),+) => {
        if cfg!(debug_assertions) {
            std::dbg!($($xs),+)
        } else {
            ($($xs),+)
        }
    }
}

fn main() {
    input!{
        a: usize,
        b: usize,
        c: usize,
        n: usize,
    }

    let m = 51;
    let mut dp = vec![vec![vec![vec![[0.0; 3]; 51]; 51]; 51]; 51];

    for i in 1..=n {
        for j in 0..=a {
            for k in 0..=b {
                for l in 0..=c {
                    if j+k+l <= 1 { continue; }
                    let t = ((j+k+l) * (j+k+l-1) / 2) as f64;

                    let p = if j > 0 { (j*(j-1)/2) as f64 / t } else { 0.0 };
                    let q = if k > 0 { (k*(k-1)/2) as f64 / t } else { 0.0 };
                    let r = if l > 0 { (l*(l-1)/2) as f64 / t } else { 0.0 };
                    // eprintln!("{} {} {} {} {} {} {}", i, j, k, l, p, q, r);

                    for h in 0..3 {
                        dp[i][j][k][l][h] += (1.0-p-q-r) * dp[i-1][j][k][l][h];
                    }

                    if j > 0 {
                        dp[i][j][k][l][0] += p * (dp[i-1][j-1][k][l][0] + 1.0);
                        dp[i][j][k][l][1] += p * dp[i-1][j-1][k][l][1];
                        dp[i][j][k][l][2] += p * dp[i-1][j-1][k][l][2];
                    }
                    if k > 0 {
                        dp[i][j][k][l][0] += q * dp[i-1][j][k-1][l][0];
                        dp[i][j][k][l][1] += q * (dp[i-1][j][k-1][l][1] + 1.0);
                        dp[i][j][k][l][2] += q * dp[i-1][j][k-1][l][2];
                    }
                    if l > 0 {
                        dp[i][j][k][l][0] += r * dp[i-1][j][k][l-1][0];
                        dp[i][j][k][l][1] += r * dp[i-1][j][k][l-1][1];
                        dp[i][j][k][l][2] += r * (dp[i-1][j][k][l-1][2] + 1.0);
                    }
                }
            }
        }
    }

    println!("{} {} {}", dp[n][a][b][c][0], dp[n][a][b][c][1], dp[n][a][b][c][2]);
}

pub mod proconio {
    #[macro_export]
    macro_rules! input{
        (sc=$sc:expr,$($r:tt)*)=>{
            input_inner!{$sc,$($r)*}
        };
        ($($r:tt)*)=>{
            let mut sc=proconio::Scanner::new(std::io::stdin().lock());
            input_inner!{sc,$($r)*}
        };
    }

    #[macro_export]
    macro_rules! input_inner{
        ($sc:expr)=>{};
        ($sc:expr,)=>{};
        ($sc:expr,$var:ident:$t:tt$($r:tt)*)=>{
            let $var=read_value!($sc,$t);
            input_inner!{$sc $($r)*}
        };
    }

    #[macro_export]
    macro_rules! read_value{
        ($sc:expr,($($t:tt),*))=>{
            ($(read_value!($sc,$t)),*)
        };
        ($sc:expr,[$t:tt;$len:expr])=>{
            (0..$len).map(|_|read_value!($sc,$t)).collect::<Vec<_>>()
        };
        ($sc:expr,Chars)=>{read_value!($sc,String).chars().collect::<Vec<char>>()};
        ($sc:expr,Usize1)=>{read_value!($sc,usize)-1};
        ($sc:expr,$t:ty)=>{$sc.next::<$t>()};
    }
    pub struct Scanner {
        s: Box<str>,
        input: std::iter::Peekable<std::str::SplitAsciiWhitespace<'static>>,
    }
    impl Scanner {
        pub fn new<R: std::io::Read>(mut reader: R) -> Self {
            let s = {
                let mut s = String::new();
                reader.read_to_string(&mut s).unwrap();
                s.into_boxed_str()
            };
            let mut sc = Scanner {
                s,
                input: "".split_ascii_whitespace().peekable(),
            };
            use std::mem;
            let s: &'static str = unsafe { mem::transmute(&*sc.s) };
            sc.input = s.split_ascii_whitespace().peekable();
            sc
        }
        #[inline]
        pub fn next<T: std::str::FromStr>(&mut self) -> T
        where
            T::Err: std::fmt::Debug,
        {
            self.input
                .next()
                .unwrap()
                .parse::<T>()
                .expect("Parse error")
        }
    }
}
0