Submission #3609444

---

Source Code Expand

macro_rules! input {
    (source = $s:expr, $($r:tt)*) => {
        let mut iter = $s.split_whitespace();
        let mut next = || { iter.next().unwrap() };
        input_inner!{next, $($r)*}
    };
    ($($r:tt)*) => {
        let stdin = std::io::stdin();
        let mut bytes = std::io::Read::bytes(std::io::BufReader::new(stdin.lock()));
        let mut next = move || -> String{
            bytes
                .by_ref()
                .map(|r|r.unwrap() as char)
                .skip_while(|c|c.is_whitespace())
                .take_while(|c|!c.is_whitespace())
                .collect()
        };
        input_inner!{next, $($r)*}
    };
}

macro_rules! input_inner {
    ($next:expr) => {};
    ($next:expr, ) => {};

    ($next:expr, $var:ident : $t:tt $($r:tt)*) => {
        let $var = read_value!($next, $t);
        input_inner!{$next $($r)*}
    };
}

macro_rules! read_value {
    ($next:expr, ( $($t:tt),* )) => {
        ( $(read_value!($next, $t)),* )
    };

    ($next:expr, [ $t:tt ; $len:expr ]) => {
        (0..$len).map(|_| read_value!($next, $t)).collect::<Vec<_>>()
    };

    ($next:expr, chars) => {
        read_value!($next, String).chars().collect::<Vec<char>>()
    };

    ($next:expr, usize1) => {
        read_value!($next, usize) - 1
    };

    ($next:expr, $t:ty) => {
        $next().parse::<$t>().expect("Parse error")
    };
}

fn main() {
    input!{
        h: usize,    // 縦棒の長さ - 1
        w: usize,    // 縦棒の本数
        k: usize,    // 最終的に辿り着かなければならない縦棒は左から何番目か
    }

    let m = 1_000_000_007;

    // a[h][x] h = 上からh列目を通過した直後, x = 左からx番目の縦棒にいる
    // その場合の数
    let mut a = [[0u64; 1000]; 1000];

    a[0][0] = 1;

    for h in 1..(h+1) {
        for bars in 0u8..(1 << (w-1)) {
            // 連結があるならスキップ
            if bars & (bars << 1) != 0 {
                continue;
            }

            for x in 0..w {
                let left = x > 0 && (bars & (1 << (x-1)) != 0);
                let right = x < w-1 && (bars & (1 << x) != 0);

                // println!("h = {}, x = {}", h, x);
                // println!("bar = {:b}, left = {}, right = {}", bars, left, right);
                // 左へ行くか
                if left {
                    a[h][x-1] += a[h-1][x];
                    a[h][x-1] %= m;
                }
                // 右へ行くか
                else if right {
                    a[h][x+1] += a[h-1][x];
                    a[h][x+1] %= m;
                }
                // まっすぐ降りるか
                else {
                    a[h][x] += a[h-1][x];
                    a[h][x] %= m;
                }

                // for x in 0..w {
                    // print!("{} ", a[h][x]);
                // }
                // println!("");

            }
        }

        // println!("h = {}, result ...", h);
        // for x in 0..w {
            // print!("{} ", a[h][x]);
        // }
        // println!("");
    }

    println!("{}", a[h][k-1] % m);
}

Submission Info

Judge Result