結果

問題 No.2411 Reverse Directions
ユーザー koba-e964koba-e964
提出日時 2023-08-15 02:00:23
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 29 ms / 2,000 ms
コード長 5,356 bytes
コンパイル時間 15,600 ms
コンパイル使用メモリ 376,548 KB
実行使用メモリ 12,032 KB
最終ジャッジ日時 2024-11-23 04:55:58
合計ジャッジ時間 16,008 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 1 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 6 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 AC 10 ms
7,808 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 1 ms
5,248 KB
testcase_10 AC 20 ms
12,032 KB
testcase_11 AC 1 ms
5,248 KB
testcase_12 AC 29 ms
8,064 KB
testcase_13 AC 1 ms
5,248 KB
testcase_14 AC 9 ms
5,248 KB
testcase_15 AC 3 ms
5,248 KB
testcase_16 AC 3 ms
5,248 KB
testcase_17 AC 10 ms
5,248 KB
testcase_18 AC 4 ms
5,248 KB
testcase_19 AC 1 ms
5,248 KB
testcase_20 AC 8 ms
5,248 KB
testcase_21 AC 2 ms
5,248 KB
testcase_22 AC 20 ms
5,632 KB
testcase_23 AC 1 ms
5,248 KB
testcase_24 AC 28 ms
6,528 KB
testcase_25 AC 1 ms
5,248 KB
testcase_26 AC 2 ms
5,248 KB
testcase_27 AC 1 ms
5,248 KB
testcase_28 AC 16 ms
5,760 KB
testcase_29 AC 29 ms
8,576 KB
testcase_30 AC 18 ms
7,040 KB
testcase_31 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::collections::*;
use std::io::{Write, BufWriter};
// https://qiita.com/tanakh/items/0ba42c7ca36cd29d0ac8
macro_rules! input {
    ($($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 ; $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"));
}

const INF: i32 = 1 << 28;

fn make_dist(s: &[Vec<char>], dist: &mut [Vec<(i32, u8)>]) {
    let h = s.len();
    let w = s[0].len();
    let dxy = [(0i32, 1i32), (0, -1), (1, 0), (-1, 0)];
    let mut que = VecDeque::new();
    que.push_back((0, 0, 0));
    while let Some((d, x, y)) = que.pop_front() {
        if dist[x][y].0 <= d { continue; }
        dist[x][y].0 = d;
        for &(dx, dy) in &dxy {
            let nx = (x as i32 + dx) as usize;
            let ny = (y as i32 + dy) as usize;
            if nx >= h || ny >= w { continue; }
            if s[nx][ny] == '#' { continue; }
            que.push_back((d + 1, nx, ny));
        }
    }
    for i in 0..h {
        for j in 0..w {
            if s[i][j] == '#' { continue; }
            if i + 1 < h && i > 0 && s[i + 1][j] == '.' && s[i - 1][j] == '.' {
                dist[i][j].1 |= 1;
            }
            if j + 1 < w && j > 0 && s[i][j + 1] == '.' && s[i][j - 1] == '.' {
                dist[i][j].1 |= 2;
            }
        }
    }
}

fn recover_path(dist: &[Vec<(i32, u8)>], mut x: usize, mut y: usize, d: usize) -> Vec<char> {
    assert_eq!((dist[x][y].0 as usize + d) % 2, 0);
    assert!(dist[x][y].0 as usize <= d);
    let oldx = x;
    let oldy = y;
    let h = dist.len();
    let w = dist[0].len();
    let mut ans = vec![];
    let mut cur = dist[x][y].0;
    let dxy = [(0i32, 1i32, 'L'), (0, -1, 'R'), (1, 0, 'U'), (-1, 0, 'D')];
    while cur > 0 {
        let mut nxt = (h, w);
        for &(dx, dy, ch) in &dxy {
            let nx = (x as i32 + dx) as usize;
            let ny = (y as i32 + dy) as usize;
            if nx >= h || ny >= w { continue; }
            if dist[nx][ny].0 == cur - 1 {
                nxt = (nx, ny);
                ans.push(ch);
                break;
            }
        }
        cur -= 1;
        x = nxt.0;
        y = nxt.1;
    }
    ans.reverse();
    for i in 0..d - dist[oldx][oldy].0 as usize {
        if (dist[oldx][oldy].1 & 2) != 0 {
            ans.push(['R', 'L'][i % 2]);
        } else {
            ans.push(['D', 'U'][i % 2]);
        }
    }
    ans
}

// https://yukicoder.me/problems/no/2411 (3)
// L 番目から R 番目までは偶数長であり、なおかつ delta が 0 である必要がある。
// これが達成できるためには、L 番目の直前に縦 ３ マスあるいは横 3 マスの空きの中央に幾つ用があり、
// そのあと R 番目の直後から K 番目に (H, W) に到着できる必要がある。
fn main() {
    let out = std::io::stdout();
    let mut out = BufWriter::new(out.lock());
    macro_rules! puts {($($format:tt)*) => (let _ = write!(out,$($format)*););}
    input! {
        h: usize, w: usize, k: usize, l: usize1, r: usize,
        s: [chars; h],
    }
    if (r - l) % 2 != 0 || (h + w + k) % 2 != 0 {
        puts!("No\n");
        return;
    }
    if h + w - 2 > k - (r - l) {
        puts!("No\n");
        return;
    }
    let mut dist1 = vec![vec![(INF, 0); w]; h];
    make_dist(&s, &mut dist1);
    let mut srev = s.clone();
    srev.reverse();
    for s in &mut srev {
        s.reverse();
    }
    let mut dist2 = vec![vec![(INF, 0); w]; h];
    make_dist(&srev, &mut dist2);
    let mut pivot = (h, w);
    'outer:
    for i in 0..h {
        for j in 0..w {
            if dist1[i][j].1 != 0 && (dist1[i][j].0 + l as i32) % 2 == 0
                && dist1[i][j].0 <= l as i32
                && dist2[h - 1 - i][w - 1 - j].0 <= (k - r) as i32 {
                assert_eq!((dist2[h - 1 - i][w - 1 - j].0 + (k - r) as i32) % 2, 0);
                pivot = (i, j);
                break 'outer;
            }
        }
    }
    if pivot == (h, w) {
        puts!("No\n");
        return;
    }
    let mut ans = recover_path(&dist1, pivot.0, pivot.1, l);
    for i in l..r {
        if (dist1[pivot.0][pivot.1].1 & 2) != 0 {
            ans.push(['R', 'L'][(i - l) % 2]);
        } else {
            ans.push(['D', 'U'][(i - l) % 2]);
        }
    }
    let mut tmp = recover_path(&dist2, h - 1 - pivot.0, w - 1 - pivot.1, k - r);
    tmp.reverse();
    ans.extend_from_slice(&tmp);
    puts!("Yes\n");
    puts!("{}\n", ans.into_iter().collect::<String>());
}
0