CSES - Datatähti 2022 loppu

Submission details
Task:	Sokkelo
Sender:	xnor
Submission time:	2022-01-22 14:56:10 +0200
Language:	Rust
Status:	READY
Result:	100

Feedback
group	verdict	score
#1	ACCEPTED	28
#2	ACCEPTED	72

Test results
test	verdict	time	group
#1	ACCEPTED	0.01 s	1, 2	details
#2	ACCEPTED	0.01 s	1, 2	details
#3	ACCEPTED	0.01 s	1, 2	details
#4	ACCEPTED	0.23 s	2	details
#5	ACCEPTED	0.24 s	2	details
#6	ACCEPTED	0.22 s	2	details
#7	ACCEPTED	0.01 s	1, 2	details
#8	ACCEPTED	0.17 s	2	details
#9	ACCEPTED	0.17 s	2	details
#10	ACCEPTED	0.01 s	1, 2	details
#11	ACCEPTED	0.28 s	2	details
#12	ACCEPTED	0.01 s	1, 2	details
#13	ACCEPTED	0.01 s	2	details
#14	ACCEPTED	0.01 s	1, 2	details
#15	ACCEPTED	0.12 s	2	details
#16	ACCEPTED	0.01 s	2	details
#17	ACCEPTED	0.01 s	2	details

Code

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, &mut stdin);

    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, stdin: &mut impl BufRead) -> Vec<Vec<u8>> {
    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());

        if maze.last().unwrap().len() < m {
            return Vec::new();
        }
    }

    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 details

Test 1

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### #A.................# #..................# #..................# ...` view save

correct output
`1` view save

user output
`1` view save

Test 2

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### #A.................# #..................# #..................# ...` view save

correct output
`2` view save

user output
`2` view save

Test 3

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### #A.................# #..................# #..................# ...` view save

correct output
`9` view save

user output
`9` view save

Test 4

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`1` view save

user output
`1` view save

Test 5

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`2` view save

user output
`2` view save

Test 6

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`335` view save

user output
`335` view save

Test 7

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### #####.############## ###.....############ ##.......########### ...` view save

correct output
`10` view save

user output
`10` view save

Test 8

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`436` view save

user output
`436` view save

Test 9

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`2` view save

user output
`2` view save

Test 10

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### #B................## #################.## #################.## ...` view save

correct output
`2` view save

user output
`2` view save

Test 11

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`2` view save

user output
`2` view save

Test 12

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### ##########A######### ##########.######### ##########.######### ...` view save

correct output
`2` view save

user output
`2` view save

Test 13

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`2` view save

user output
`2` view save

Test 14

Group: 1, 2

Verdict: ACCEPTED

input
`20 20 #################### ##########A######### ##########.######### ##########.######### ...` view save

correct output
`12` view save

user output
`12` view save

Test 15

Group: 2

Verdict: ACCEPTED

input
`1000 1000 ##############################...` view save

correct output
`502` view save

user output
`502` view save

Test 16

Group: 2

Verdict: ACCEPTED

input
`3 1000 ##############################...` view save

correct output
`1` view save

user output
`1` view save

Test 17

Group: 2

Verdict: ACCEPTED

input
`1000 3 ### #A# #.# #.# ...` view save

correct output
`1` view save

user output
`1` view save

Datatähti 2022 loppu

Sokkelo

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Tasks

Submissions (xnor)