被围绕的区域

struct Solution{

}

impl Solution {
    fn new() -> Self {
        Solution{}
    }

    /**
    * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
    *
    * 
        * @param board char字符型二维数组 
        * @return char字符型二维数组
    */
    pub fn surroundedArea(&self, board: Vec<Vec<char>>) -> Vec<Vec<char>> {
        // write code here
        use std::iter;
        let mut new_board = Vec::new();
        let n = board[0].len();
        new_board.push(iter::repeat('O').take(n + 2).collect::<Vec<_>>());
        for mut i in board.into_iter() {
            let mut v = vec!['O'];
            v.append(&mut i);
            v.push('O');
            new_board.push(v);
        }
        new_board.push(iter::repeat('O').take(n + 2).collect::<Vec<_>>());
        let mut board = new_board;
        let m = board.len();
        let n = board[0].len();
        let is_valid = |(x, y): (isize, isize)| {
            x >= 0 && y >= 0 && x < m as isize && y < n as isize
        };

        depth_first_traversal(&mut board, (0, 0), &is_valid);
        
        for i in 0 .. m {
            for j in 0 .. n {
                board[i][j] = if board[i][j] == 'O' {'X'} else {board[i][j]};
            }
        }
        for i in 0 .. m {
            for j in 0 .. n {
                board[i][j] = if board[i][j] == 'Z' {'O'} else {board[i][j]};
            }
        }
        // println!("{:?}", board);
        
        board[1 .. m - 1].into_iter().map(|i| i[1 .. n - 1].into_iter().map(|c| *c).collect::<Vec<_>>()).collect::<Vec<Vec<char>>>()
    }
}

fn depth_first_traversal<F>(board: &mut Vec<Vec<char>>, pos: (isize, isize), is_valid: &F) -> () where F: Fn((isize, isize)) -> bool {
    if is_valid(pos) && board[pos.0 as usize][pos.1 as usize] == 'O' {
        board[pos.0 as usize][pos.1 as usize] = 'Z';
        let dirs = [(0, 1), (0, -1), (-1, 0), (1, 0)];
        for dir in dirs.iter() {
            depth_first_traversal(board, (pos.0 + dir.0, pos.1 + dir.1), is_valid);
        }
    }
}