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
58
59
|
use std::io::{self, Read};
#[derive(Debug, Clone, Copy)]
struct Hex(i32, i32, i32);
impl Hex {
fn dist(self) -> i32 {
(self.0.abs() + self.1.abs() + self.2.abs()) / 2
}
fn mov(self, dir: &str) -> Self {
let Hex(x, y, z) = self;
match dir {
"n" => Hex(x, y + 1, z - 1),
"ne" => Hex(x + 1, y, z - 1),
"se" => Hex(x + 1, y - 1, z),
"s" => Hex(x, y - 1, z + 1),
"sw" => Hex(x - 1, y, z + 1),
"nw" => Hex(x - 1, y + 1, z),
_ => unimplemented!(),
}
}
}
fn solve1(input: &str) -> i32 {
let mut hex = Hex(0, 0, 0);
for dir in input.split(',') {
hex = hex.mov(dir);
}
hex.dist()
}
fn solve2(input: &str) -> i32 {
let mut hex = Hex(0, 0, 0);
let mut max = 0;
for dir in input.split(',') {
hex = hex.mov(dir);
if hex.dist() > max {
max = hex.dist()
}
}
max
}
fn main() {
let mut input = String::new();
io::stdin().read_to_string(&mut input).unwrap();
println!("Part 1: {}", solve1(input.trim()));
println!("Part 2: {}", solve2(input.trim()));
}
#[test]
fn part1() {
assert_eq!(3, solve1("ne,ne,ne"));
assert_eq!(0, solve1("ne,ne,sw,sw"));
assert_eq!(2, solve1("ne,ne,s,s"));
assert_eq!(3, solve1("se,sw,se,sw,sw"));
}
|