blob: 31c4c5dcabea76e9ac9f7b864b61a52af018274b (
plain) (
blame)
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
|
pub fn parse(fh: std::fs::File) -> anyhow::Result<Vec<i64>> {
Ok(crate::util::parse::ints(crate::util::parse::lines(fh)).collect())
}
pub fn part1(ints: Vec<i64>) -> anyhow::Result<i64> {
for i in &ints {
for j in &ints {
if i + j == 2020 {
return Ok(i * j);
}
}
}
Err(anyhow::anyhow!("no numbers summing to 2020 found"))
}
pub fn part2(ints: Vec<i64>) -> anyhow::Result<i64> {
for i in &ints {
for j in &ints {
for k in &ints {
if i + j + k == 2020 {
return Ok(i * j * k);
}
}
}
}
Err(anyhow::anyhow!("no numbers summing to 2020 found"))
}
#[test]
fn test() {
assert_eq!(
part1(parse(crate::util::data(2020, 1).unwrap()).unwrap()).unwrap(),
445536
);
assert_eq!(
part2(parse(crate::util::data(2020, 1).unwrap()).unwrap()).unwrap(),
138688160
);
}
|