solution: day1
This commit is contained in:
97
src/bin/01.rs
Normal file
97
src/bin/01.rs
Normal file
@@ -0,0 +1,97 @@
|
|||||||
|
use hashbrown::HashSet;
|
||||||
|
|
||||||
|
enum Rotation {
|
||||||
|
Left,
|
||||||
|
Right,
|
||||||
|
}
|
||||||
|
|
||||||
|
fn rotate(direction: (isize, isize), rotation: &Rotation) -> (isize, isize) {
|
||||||
|
match rotation {
|
||||||
|
Rotation::Right => (-direction.1, direction.0),
|
||||||
|
Rotation::Left => (direction.1, -direction.0),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
enum Direction {
|
||||||
|
Move(isize),
|
||||||
|
Rotate(Rotation),
|
||||||
|
}
|
||||||
|
|
||||||
|
fn expand_directions(dv: &mut Vec<Direction>, d: &str) {
|
||||||
|
let (r, m) = d.split_at(1);
|
||||||
|
|
||||||
|
let dir = match r {
|
||||||
|
"L" => Rotation::Left,
|
||||||
|
"R" => Rotation::Right,
|
||||||
|
_ => unreachable!(),
|
||||||
|
};
|
||||||
|
|
||||||
|
dv.push(Direction::Rotate(dir));
|
||||||
|
dv.push(Direction::Move(m.parse().unwrap()));
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn part_one(input: &str) -> Option<isize> {
|
||||||
|
let mut directions = Vec::new();
|
||||||
|
|
||||||
|
for d in input.split(", ") {
|
||||||
|
expand_directions(&mut directions, d);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (mut x, mut y) = (0, 0);
|
||||||
|
let (mut dx, mut dy) = (1, 0);
|
||||||
|
|
||||||
|
for d in directions {
|
||||||
|
match d {
|
||||||
|
Direction::Rotate(r) => (dx, dy) = rotate((dx, dy), &r),
|
||||||
|
Direction::Move(n) => (x, y) = (x + dx * n, y + dy * n),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
Some(x.abs() + y.abs())
|
||||||
|
}
|
||||||
|
pub fn part_two(input: &str) -> Option<isize> {
|
||||||
|
let mut directions = Vec::new();
|
||||||
|
|
||||||
|
for d in input.split(", ") {
|
||||||
|
expand_directions(&mut directions, d);
|
||||||
|
}
|
||||||
|
|
||||||
|
let (mut x, mut y) = (0, 0);
|
||||||
|
let (mut dx, mut dy) = (1, 0);
|
||||||
|
let mut visited = HashSet::new();
|
||||||
|
|
||||||
|
for d in directions {
|
||||||
|
match d {
|
||||||
|
Direction::Rotate(r) => (dx, dy) = rotate((dx, dy), &r),
|
||||||
|
Direction::Move(n) => {
|
||||||
|
for _ in 1..=n {
|
||||||
|
(x, y) = (x + dx, y + dy);
|
||||||
|
if !visited.insert((x, y)) {
|
||||||
|
return Some(x.abs() + y.abs());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
None
|
||||||
|
}
|
||||||
|
fn main() {
|
||||||
|
let input = &aoc::read_file("inputs", 1);
|
||||||
|
aoc::solve!(1, part_one, input);
|
||||||
|
aoc::solve!(2, part_two, input);
|
||||||
|
}
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use super::*;
|
||||||
|
#[test]
|
||||||
|
fn test_part_one() {
|
||||||
|
let input = aoc::read_file("examples", 1);
|
||||||
|
assert_eq!(part_one(&input), Some(8));
|
||||||
|
}
|
||||||
|
#[test]
|
||||||
|
fn test_part_two() {
|
||||||
|
let input = aoc::read_file("examples", 1);
|
||||||
|
assert_eq!(part_two(&input), Some(4));
|
||||||
|
}
|
||||||
|
}
|
||||||
1
src/examples/01.txt
Normal file
1
src/examples/01.txt
Normal file
@@ -0,0 +1 @@
|
|||||||
|
R8, R4, R4, R8
|
||||||
Reference in New Issue
Block a user