Submission #65738144

---

Source Code Expand

use itertools::iproduct;
use proconio::{input, marker::Bytes};
use std::collections::VecDeque;

fn main() {
    input! {
        h: usize,
        w: usize,
        mut ans: [Bytes; h],
    }
    let mut queue = iproduct!(0..h, 0..w)
        .filter(|&(i, j)| ans[i][j] == b'E')
        .collect::<VecDeque<_>>();
    while let Some((i, j)) = queue.pop_front() {
        for (ni, nj, c) in [
            (i + 1, j, b'^'),
            (i.wrapping_sub(1), j, b'v'),
            (i, j + 1, b'<'),
            (i, j.wrapping_sub(1), b'>'),
        ]
        .into_iter()
        .filter(|&(ni, nj, _)| ni < h && nj < w)
        {
            if ans[ni][nj] == b'.' {
                ans[ni][nj] = c;
                queue.push_back((ni, nj));
            }
        }
    }
    for ans in &ans {
        println!("{}", String::from_utf8_lossy(ans));
    }
}

Submission Info

Judge Result