Sokkelo

warning: value assigned to `neighbors_b` is never read
   --> input/code.rs:166:9
    |
166 |         neighbors_b = new_neighbors_b;
    |         ^^^^^^^^^^^
    |
    = note: `#[warn(unused_assignments)]` on by default
    = help: maybe it is overwritten before being read?

warning: variable does not need to be mutable
   --> input/code.rs:124:13
    |
124 |         let mut new_neighbors_a = HashSet::new();
    |             ----^^^^^^^^^^^^^^^
    |             |
    |             help: remove this `mut`
    |
    = note: `#[warn(unused_mut)]` on by default

warning: variable does not need to be mutable
   --> input/code.rs:125:13
    |
125 |         let mut new_neighbors_b = HashSet::new();
    |             ----^^^^^^^^^^^^^^^
    |             |
    |             help: remove this `mut`

warning: 3 warnings emitted

use std::{
    collections::HashSet,
    io::{stdin, BufRead, BufReader},
};

fn main() {
    let mut stdin = BufReader::new(stdin());
    let mut cur_line = String::new();

    stdin.read_line(&mut cur_line).unwrap();
    let mut nums = cur_line
        .split_whitespace()
        .map(|s| s.parse::<usize>().unwrap());

    let n = nums.next().unwrap();
    let m = nums.next().unwrap();

    let maze = fill_maze(n, m);

    println!("{}", get_dist(maze, n, m));
}

fn get_dist(mut maze: Vec<Vec<u8>>, n: usize, m: usize) -> usize {
    let mut neighbors_a = HashSet::new();

    for y in 1..n - 1 {
        for x in 1..m - 1 {
            if maze[y][x] == b'A' {
                if y - 1 > 0 && maze[y - 1][x] != b'A' {
                    neighbors_a.insert((x, y - 1));
                }
                if y + 1 < n - 1 && maze[y + 1][x] != b'A' {
                    neighbors_a.insert((x, y + 1));
                }
                if x - 1 > 0 && maze[y][x - 1] != b'A' {
                    neighbors_a.insert((x - 1, y));
                }
                if x + 1 < m - 1 && maze[y][x + 1] != b'A' {
                    neighbors_a.insert((x + 1, y));
                }
            }
        }
    }

    for c in 1.. {
        let mut new_neighbors = HashSet::new();

        for &(x, y) in &neighbors_a {
            if maze[y][x] == b'B' {
                return c;
            }

            maze[y][x] = b'A';

            if y - 1 > 0 && maze[y - 1][x] != b'A' {
                new_neighbors.insert((x, y - 1));
            }
            if y + 1 < n - 1 && maze[y + 1][x] != b'A' {
                new_neighbors.insert((x, y + 1));
            }
            if x - 1 > 0 && maze[y][x - 1] != b'A' {
                new_neighbors.insert((x - 1, y));
            }
            if x + 1 < m - 1 && maze[y][x + 1] != b'A' {
                new_neighbors.insert((x + 1, y));
            }
        }

        neighbors_a = new_neighbors;
    }

    0
}

fn fill_maze(n: usize, m: usize) -> Vec<Vec<u8>> {
    let mut stdin = BufReader::new(stdin());

    let mut maze = Vec::with_capacity(n);

    for _ in 0..n {
        let mut cur_line = String::new();
        stdin.read_line(&mut cur_line).unwrap();

        maze.push(cur_line.into_bytes());
    }

    let mut neighbors_a = HashSet::new();
    let mut neighbors_b = HashSet::new();

    for y in 1..n - 1 {
        for x in 1..m - 1 {
            if maze[y][x] == b'A' {
                if maze[y - 1][x] == b'.' {
                    neighbors_a.insert((x, y - 1));
                }
                if maze[y + 1][x] == b'.' {
                    neighbors_a.insert((x, y + 1));
                }
                if maze[y][x - 1] == b'.' {
                    neighbors_a.insert((x - 1, y));
                }
                if maze[y][x + 1] == b'.' {
                    neighbors_a.insert((x + 1, y));
                }
            }
            if maze[y][x] == b'B' {
                if maze[y - 1][x] == b'.' {
                    neighbors_b.insert((x, y - 1));
                }
                if maze[y + 1][x] == b'.' {
                    neighbors_b.insert((x, y + 1));
                }
                if maze[y][x - 1] == b'.' {
                    neighbors_b.insert((x - 1, y));
                }
                if maze[y][x + 1] == b'.' {
                    neighbors_b.insert((x + 1, y));
                }
            }
        }
    }

    loop {
        let mut new_neighbors_a = HashSet::new();
        let mut new_neighbors_b = HashSet::new();

        for &(x, y) in &neighbors_a {
            maze[y][x] = b'A';

            if y > 0 && maze[y - 1][x] == b'.' && !neighbors_a.contains(&(x, y - 1)) {
                // new_neighbors_a.insert((x, y - 1));
            }
            if y < n - 1 && maze[y + 1][x] == b'.' && !neighbors_a.contains(&(x, y + 1)) {
                // new_neighbors_a.insert((x, y + 1));
            }
            if x > 0 && maze[y][x - 1] == b'.' && !neighbors_a.contains(&(x - 1, y)) {
                // new_neighbors_a.insert((x - 1, y));
            }
            if x < m - 1 && maze[y][x + 1] == b'.' && !neighbors_a.contains(&(x + 1, y)) {
                // new_neighbors_a.insert((x + 1, y));
            }
        }

        // for &(x, y) in &neighbors_b {
        //     maze[y][x] = b'B';

        //     if maze[y - 1][x] == b'.' && !neighbors_b.contains(&(x, y - 1)) {
        //         new_neighbors_b.insert((x, y - 1));
        //     }
        //     if maze[y + 1][x] == b'.' && !neighbors_b.contains(&(x, y + 1)) {
        //         new_neighbors_b.insert((x, y + 1));
        //     }
        //     if maze[y][x - 1] == b'.' && !neighbors_b.contains(&(x - 1, y)) {
        //         new_neighbors_b.insert((x - 1, y));
        //     }
        //     if maze[y][x + 1] == b'.' && !neighbors_b.contains(&(x + 1, y)) {
        //         new_neighbors_b.insert((x + 1, y));
        //     }
        // }

        if new_neighbors_a.is_empty() && new_neighbors_b.is_empty() {
            break maze;
        }

        neighbors_a = new_neighbors_a;
        neighbors_b = new_neighbors_b;
    }
}

Test 1

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 2

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 3

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 4

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 5

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 6

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 7

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 8

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 9

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 10

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 11

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 12

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 13

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 14

Group: 1, 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 15

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 16

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...

Test 17

Group: 2

Verdict: RUNTIME ERROR

thread 'main' panicked at 'index out of bounds: the len is 0 but the index is 1', input/co...