-
Notifications
You must be signed in to change notification settings - Fork 0
/
01.rs
98 lines (89 loc) · 2.1 KB
/
01.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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
use std::collections::HashSet;
static INPUT_TXT: &str = include_str!("../../input/01.txt");
fn main() {
println!("Part 1: {}", part_1(INPUT_TXT));
println!("Part 2: {}", part_2(INPUT_TXT));
}
fn parse_input(input: &str) -> Vec<i64> {
input
.trim()
.lines()
.map(|num| {
let mut chars = num.chars();
match chars.next() {
Some('-') => -(chars.as_str().parse::<i64>().unwrap()),
Some('+') => chars.as_str().parse::<i64>().unwrap(),
_ => unreachable!(),
}
})
.collect()
}
fn part_1(input: &str) -> i64 {
parse_input(input).into_iter().sum()
}
fn part_2(input: &str) -> i64 {
let nums = parse_input(input);
let mut counter = 0;
let mut visted: HashSet<i64> = vec![0i64].into_iter().collect();
'outer: loop {
for num in &nums {
counter += num;
if visted.contains(&counter) {
break 'outer counter;
}
visted.insert(counter);
}
}
}
#[cfg(test)]
mod day_1_tests {
use super::*;
static INPUT_1: &str = "+1
+1
+1";
static INPUT_2: &str = "+1
+1
-2";
static INPUT_3: &str = "-1
-2
-3";
static INPUT_4: &str = "+1
-1";
static INPUT_5: &str = "+3
+3
+4
-2
-4";
static INPUT_6: &str = "-6
+3
+8
+5
-6";
static INPUT_7: &str = "+7
+7
-2
-7
-4";
#[test]
fn test_part_1() {
assert_eq!(part_1(INPUT_1), 3);
assert_eq!(part_1(INPUT_2), 0);
assert_eq!(part_1(INPUT_3), -6);
assert_eq!(part_1(INPUT_4), 0);
assert_eq!(part_1(INPUT_5), 4);
assert_eq!(part_1(INPUT_6), 4);
assert_eq!(part_1(INPUT_7), 1);
assert_eq!(part_1(INPUT_TXT), 590);
}
#[test]
fn test_part_2() {
assert_eq!(part_1(INPUT_1), 3);
assert_eq!(part_1(INPUT_2), 0);
assert_eq!(part_1(INPUT_3), -6);
assert_eq!(part_2(INPUT_4), 0);
assert_eq!(part_2(INPUT_5), 10);
assert_eq!(part_2(INPUT_6), 5);
assert_eq!(part_2(INPUT_7), 14);
assert_eq!(part_2(INPUT_TXT), 83445);
}
}