結果

問題 No.1638 Robot Maze
コンテスト
ユーザー Strorkis
提出日時 2021-08-06 22:49:48
言語 Rust
(1.83.0 + proconio)
結果
AC  
実行時間 5 ms / 2,000 ms
コード長 2,831 bytes
コンパイル時間 12,857 ms
コンパイル使用メモリ 378,608 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-17 03:02:57
合計ジャッジ時間 13,965 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 49
権限があれば一括ダウンロードができます

ソースコード

diff # 

pub mod io {
    use std::io::{BufRead, ErrorKind};

    pub fn scan<R: BufRead>(r: &mut R) -> Vec<u8> {
        let mut res = Vec::new();
        loop {
            let buf = match r.fill_buf() {
                Ok(buf) => buf,
                Err(e) if e.kind() == ErrorKind::Interrupted => continue,
                Err(e) => panic!("{}", e), 
            };
            let (done, used, buf) = {
                match buf.iter().position(u8::is_ascii_whitespace) {
                    Some(i) => (i > 0 || res.len() > 0, i + 1, &buf[..i]),
                    None => (buf.is_empty(), buf.len(), buf),
                }
            };
            res.extend_from_slice(buf);
            r.consume(used);
            if done { return res; }
        }
    }

    #[macro_export]
    macro_rules! scan {
        ($r:expr, [$t:tt; $n:expr]) => {
            (0..$n).map(|_| scan!($r, $t)).collect::<Vec<_>>()
        };
        ($r:expr, [$t:tt]) => {
            scan!($r, [$t; scan!($r, usize)])
        };
        ($r:expr, ($($t:tt),*)) => {
            ($(scan!($r, $t)),*)
        };
        ($r:expr, Usize1) => {
            scan!($r, usize) - 1
        };
        ($r:expr, Bytes) => {
            io::scan($r)
        };
        ($r:expr, String) => {
            String::from_utf8(scan!($r, Bytes)).unwrap()
        };
        ($r:expr, $t:ty) => {
            scan!($r, String).parse::<$t>().unwrap()
        };
    }

    #[macro_export]
    macro_rules! input {
        ($($($v:ident)* : $t:tt),* $(,)?) => {
            let stdin = std::io::stdin();
            let ref mut reader = std::io::BufReader::new(stdin.lock());
            $(let $($v)* = scan!(reader, $t);)*
        };
    }
}

use std::collections::VecDeque;

const INF: u64 = 1 << 60;

fn main() {
    input! {
        h: usize, w: usize,
        u: u64, d: u64, r: u64, l: u64, k: u64, p: u64,
        x_s: Usize1, y_s: Usize1, x_t: Usize1, y_t: Usize1,
        c: [Bytes; h],
    }

    let mut queue = VecDeque::new();
    queue.push_back((x_s, y_s, 0));
    let mut dist = vec![vec![INF; w]; h];
    dist[x_s][y_s] = 0;
    while let Some((x, y, now_cost)) = queue.pop_front() {
        if dist[x][y] < now_cost {
            continue;
        }
        for &(dx, dy, cost) in &[(!0, 0, u), (1, 0, d), (0, 1, r), (0, !0, l)] {
            let (nx, ny) = (x.wrapping_add(dx), y.wrapping_add(dy));
            if nx >= h || ny >= w || c[nx][ny] == b'#' {
                continue;
            }
            let next_cost = now_cost + if c[nx][ny] == b'@' { cost + p } else { cost };
            if next_cost < dist[nx][ny] {
                dist[nx][ny] = next_cost;
                queue.push_back((nx, ny, next_cost));
            }
        }
    }

    if dist[x_t][y_t] <= k {
        println!("Yes");
    } else {
        println!("No");
    }
}
0