-
-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathmain.rs
57 lines (54 loc) · 1.78 KB
/
main.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
pub fn main() {
let map = include_bytes!("../input.txt");
let width = map.iter().position(|&b| b == b'\n').unwrap();
let start = map.iter().position(|&b| b == b'S').unwrap();
let (mut pos, mut dir) = {
if matches!(map[start - width - 1], b'|' | b'7' | b'F') {
(start - width - 1, Dir::Up)
} else if matches!(map[start + width + 1], b'|' | b'L' | b'J') {
(start + width + 1, Dir::Down)
} else {
(start - 1, Dir::Left)
}
};
println!(
"{}",
(1 + std::iter::repeat(())
.position(|_| unsafe {
match (map.get_unchecked(pos), dir) {
(b'|', Dir::Down) => pos += width + 1,
(b'|', Dir::Up) => pos -= width + 1,
(b'-', Dir::Left) => pos -= 1,
(b'-', Dir::Right) => pos += 1,
(b'L', Dir::Down) | (b'F', Dir::Up) => {
pos += 1;
dir = Dir::Right;
}
(b'L', Dir::Left) | (b'J', Dir::Right) => {
pos -= width + 1;
dir = Dir::Up;
}
(b'7', Dir::Up) | (b'J', Dir::Down) => {
pos -= 1;
dir = Dir::Left;
}
(b'7', Dir::Right) | (b'F', Dir::Left) => {
pos += width + 1;
dir = Dir::Down;
}
(b'S', _) => return true,
(_, _) => unreachable!(),
}
false
})
.unwrap())
/ 2
);
}
#[derive(Debug, Clone, Copy)]
pub enum Dir {
Up,
Down,
Left,
Right,
}