Add D10P1

This commit is contained in:
kageru 2019-12-10 19:59:52 +01:00
parent 558212023d
commit 93a450ca71
Signed by: kageru
GPG Key ID: 8282A2BEA4ADA3D2

View File

@ -1,20 +1,64 @@
use std::collections::HashSet;
use std::io::{self, BufRead};
#[derive(Clone)]
struct Asteroid {
x: i64,
y: i64,
visible_asteroids: HashSet<i64>,
}
impl Asteroid {
pub fn new(x: usize, y: usize) -> Self {
Self {
x: x as i64,
y: y as i64,
visible_asteroids: HashSet::new(),
}
}
fn compute_visibles(mut self, all_asteroids: &Vec<Asteroid>) -> Self {
// Technically incorrect because we add ourself here, but since the asteroid with most
// visibles has some legitimate asteroid at an angle of 0°, this doesn’t actually matter.
for ast in all_asteroids {
self.visible_asteroids
.insert(calculate_angle(self.x - ast.x, self.y - ast.y));
}
self
}
}
fn main() {
println!("Hello, world!");
let input: Vec<Asteroid> = io::stdin()
.lock()
.lines()
.enumerate()
.flat_map(|(y, l)| {
l.unwrap()
.chars()
.enumerate()
.filter(|(_, c)| c == &'#')
.map(move |(x, _)| Asteroid::new(x, y))
.collect::<Vec<_>>()
})
.collect();
let part1 = input
.clone()
.into_iter()
.map(|ast| ast.compute_visibles(&input))
.max_by_key(|ast| ast.visible_asteroids.len())
.unwrap();
println!("Part 1: {}", part1.visible_asteroids.len());
}
fn calculate_angle(x_offset: i32, y_offset: i32) -> f32 {
return ((y_offset as f32).atan2(x_offset as f32).to_degrees() * 1000.0).round();
fn calculate_angle(x_offset: i64, y_offset: i64) -> i64 {
return ((y_offset as f64).atan2(x_offset as f64).to_degrees() * 1000.0).round() as i64;
}
fn parse_input(raw: std::str::Lines) -> HashSet<(i32, i32)> {
unimplemented!()
}
fn part1(asteroids: &HashSet<(i32, i32)>) -> (i32, i32) {
unimplemented!()
}
//fn part1(asteroids: &HashSet<(i32, i32)>) -> Asteroid {
//unimplemented!()
//}
#[cfg(test)]
mod tests {
@ -22,103 +66,105 @@ mod tests {
#[test]
fn test_angle_calculation() {
assert_eq!(calculate_angle(1, 1), 45_000.0);
assert_eq!(calculate_angle(1, 0), 0.0);
assert_eq!(calculate_angle(0, -1), -90_000.0);
assert_eq!(calculate_angle(2, 1), 26_565.0);
assert_eq!(calculate_angle(1, 1), 45_000);
assert_eq!(calculate_angle(1, 0), 0);
assert_eq!(calculate_angle(0, -1), -90_000);
assert_eq!(calculate_angle(2, 1), 26_565);
assert_eq!(calculate_angle(1, 1), calculate_angle(3, 3));
assert_eq!(calculate_angle(-2, -2), -135_000.0);
assert_eq!(calculate_angle(-2, 2), 135_000.0);
assert_eq!(calculate_angle(-2, -2), -135_000);
assert_eq!(calculate_angle(-2, 2), 135_000);
assert_ne!(calculate_angle(1, 1), calculate_angle(3, -3));
}
#[test]
fn test_part_1() {
assert_eq!(
part1(&parse_input(
".#..#
.....
#####
....#
...##"
.lines()
)),
(3, 4)
);
/*
#[test]
fn test_part_1() {
assert_eq!(
part1(&parse_input(
".#..#
.....
#####
....#
...##"
.lines()
)).visible_asteroids,
HashSet::new()
);
assert_eq!(
part1(&parse_input(
"#.#...#.#.
.###....#.
.#....#...
##.#.#.#.#
....#.#.#.
.##..###.#
..#...##..
..##....##
......#...
.####.###."
.lines()
)),
(1, 2)
);
assert_eq!(
part1(&parse_input(
".#..#..###
####.###.#
....###.#.
..###.##.#
##.##.#.#.
....###..#
..#.#..#.#
#..#.#.###
.##...##.#
.....#.#.."
.lines()
)),
(6, 3)
);
assert_eq!(
part1(&parse_input(
"......#.#.
#..#.#....
..#######.
.#.#.###..
.#..#.....
..#....#.#
#..#....#.
.##.#..###
##...#..#.
.#....####"
.lines()
)),
(5, 8)
);
assert_eq!(
part1(&parse_input(
".#..##.###...#######
##.############..##.
.#.######.########.#
.###.#######.####.#.
#####.##.#.##.###.##
..#####..#.#########
####################
#.####....###.#.#.##
##.#################
#####.##.###..####..
..######..##.#######
####.##.####...##..#
.#####..#.######.###
##...#.##########...
#.##########.#######
.####.#.###.###.#.##
....##.##.###..#####
.#.#.###########.###
#.#.#.#####.####.###
###.##.####.##.#..##"
.lines()
)),
(11, 13)
);
}
assert_eq!(
part1(&parse_input(
"#.#...#.#.
.###....#.
.#....#...
##.#.#.#.#
....#.#.#.
.##..###.#
..#...##..
..##....##
......#...
.####.###."
.lines()
)),
(1, 2)
);
assert_eq!(
part1(&parse_input(
".#..#..###
####.###.#
....###.#.
..###.##.#
##.##.#.#.
....###..#
..#.#..#.#
#..#.#.###
.##...##.#
.....#.#.."
.lines()
)),
(6, 3)
);
assert_eq!(
part1(&parse_input(
"......#.#.
#..#.#....
..#######.
.#.#.###..
.#..#.....
..#....#.#
#..#....#.
.##.#..###
##...#..#.
.#....####"
.lines()
)),
(5, 8)
);
assert_eq!(
part1(&parse_input(
".#..##.###...#######
##.############..##.
.#.######.########.#
.###.#######.####.#.
#####.##.#.##.###.##
..#####..#.#########
####################
#.####....###.#.#.##
##.#################
#####.##.###..####..
..######..##.#######
####.##.####...##..#
.#####..#.######.###
##...#.##########...
#.##########.#######
.####.#.###.###.#.##
....##.##.###..#####
.#.#.###########.###
#.#.#.#####.####.###
###.##.####.##.#..##"
.lines()
)),
(11, 13)
);
}
*/
}