結果

問題 No.162 8020運動
ユーザー koba-e964koba-e964
提出日時 2017-02-07 09:10:31
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 1,097 ms / 5,000 ms
コード長 3,169 bytes
コンパイル時間 22,298 ms
コンパイル使用メモリ 378,740 KB
実行使用メモリ 7,168 KB
最終ジャッジ日時 2024-06-06 17:23:42
合計ジャッジ時間 35,265 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 59 ms
6,812 KB
testcase_01 AC 539 ms
6,940 KB
testcase_02 AC 717 ms
6,940 KB
testcase_03 AC 931 ms
7,040 KB
testcase_04 AC 939 ms
7,168 KB
testcase_05 AC 962 ms
7,040 KB
testcase_06 AC 955 ms
7,040 KB
testcase_07 AC 962 ms
7,040 KB
testcase_08 AC 947 ms
7,040 KB
testcase_09 AC 52 ms
6,944 KB
testcase_10 AC 709 ms
6,944 KB
testcase_11 AC 540 ms
6,940 KB
testcase_12 AC 253 ms
6,944 KB
testcase_13 AC 1,010 ms
7,040 KB
testcase_14 AC 102 ms
6,940 KB
testcase_15 AC 103 ms
6,940 KB
testcase_16 AC 310 ms
6,944 KB
testcase_17 AC 149 ms
6,944 KB
testcase_18 AC 1,097 ms
6,944 KB
testcase_19 AC 663 ms
6,944 KB
testcase_20 AC 886 ms
6,940 KB
testcase_21 AC 748 ms
6,944 KB
testcase_22 AC 719 ms
6,940 KB
testcase_23 AC 787 ms
6,944 KB
testcase_24 AC 770 ms
6,944 KB
testcase_25 AC 810 ms
6,940 KB
testcase_26 AC 51 ms
6,940 KB
testcase_27 AC 523 ms
6,940 KB
testcase_28 AC 892 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#[allow(unused_imports)]
use std::cmp::*;
#[allow(unused_imports)]
use std::collections::*;
use std::io::*;
#[allow(dead_code)]
fn getline() -> String {
    let mut ret = String::new();
    std::io::stdin().read_line(&mut ret).ok();
    return ret;
}
fn get_word() -> String {
    let mut stdin = std::io::stdin();
    let mut u8b: [u8; 1] = [0];
    loop {
        let mut buf: Vec<u8> = Vec::with_capacity(16);
        loop {
            let res = stdin.read(&mut u8b);
            if res.is_err() || res.ok().unwrap() == 0 || u8b[0] <= ' ' as u8 {
                break;
            } else {
                buf.push(u8b[0]);
            }
        }
        if buf.len() >= 1 {
            let ret = std::string::String::from_utf8(buf).unwrap();
            return ret;
        }
    }
}
fn parse<T: std::str::FromStr>(s: &str) -> T { s.parse::<T>().ok().unwrap() }

#[allow(dead_code)]
fn get<T: std::str::FromStr>() -> T { parse(&get_word()) }

fn main() {
    let n = 80 - get::<usize>();
    let p0: f64 = get::<f64>() / 100.0;
    let p1: f64 = get::<f64>() / 100.0;
    let p2: f64 = get::<f64>() / 100.0;
    const B: usize = 14;
    let mut decay = vec![vec![0.0; B]; 1 << B];
    for bits in 0 .. 1 << B {
        for j in 0 .. B {
            if (bits & 1 << j) == 0 { continue; }
            let p;
            let mut adj = 0;
            if j == 0 || (j >= 1 && (bits & 1 << (j - 1)) == 0) {
                adj += 1;
            }
            if j == B - 1 || (j < B - 1 && (bits & 1 << (j + 1)) == 0) {
                adj += 1;
            }
            p = match adj {
                0 => p2,
                1 => p1,
                2 => p0,
                _ => panic!(),
            } + 1e-8;
            decay[bits][j] = p;
        }
    }
    let mut dp = vec![vec![0.0; 1 << B]; n + 1];
    dp[0][(1 << B) - 1] = 1.0;
    for i in 0 .. n {
        for b1 in 0 .. 1 << B {
            let mut b2 = b1;
            let mut prob = 1.0;
            for j in 0 .. B {
                if (b1 & 1 << j) == 0 { continue; }
                prob *= 1.0 - decay[b1][j];
            }
            loop {
                /*
                for j in 0 .. B {
                    if (b1 & 1 << j) == 0 { continue; }
                    let p = decay[b1][j];
                    prob *= if (b2 & 1 << j) == 0 { p } else { 1.0 - p };
                }
                 */
                dp[i + 1][b2] += prob * dp[i][b1];
                if b2 == 0 {
                    break;
                }
                let newb2 = (b2 - 1) & b1;
                for j in 0 .. B {
                    if (b1 & 1 << j) == 0 { continue; }
                    if (b2 & 1 << j) == (newb2 & 1 << j) { break; }
                    if (newb2 & 1 << j) == 0 {
                        prob *= decay[b1][j] / (1.0 - decay[b1][j]);
                    } else {
                        prob *= (1.0 - decay[b1][j]) / decay[b1][j];
                    }
                }
                b2 = newb2;
            }
        }
    }
    let mut tot = 0.0;
    for i in 0 .. 1 << B {
        tot += ((i as u32).count_ones() as f64) * dp[n][i];
    }
    println!("{}", 2.0 * tot);
}
0