diff options
Diffstat (limited to '2021/day9/src/main.rs')
-rw-r--r-- | 2021/day9/src/main.rs | 67 |
1 files changed, 67 insertions, 0 deletions
diff --git a/2021/day9/src/main.rs b/2021/day9/src/main.rs new file mode 100644 index 0000000..75bf1e9 --- /dev/null +++ b/2021/day9/src/main.rs | |||
@@ -0,0 +1,67 @@ | |||
1 | use std::env; | ||
2 | use std::fs::File; | ||
3 | use std::io::{BufRead, BufReader}; | ||
4 | |||
5 | fn main() { | ||
6 | let reader = open_file(); | ||
7 | |||
8 | let mut points: Vec<Vec<u32>> = Vec::new(); | ||
9 | |||
10 | for line in reader.lines() { | ||
11 | points.push( | ||
12 | line.unwrap() | ||
13 | .trim() | ||
14 | .chars() | ||
15 | .map(|x| x.to_digit(10).unwrap()) | ||
16 | .collect(), | ||
17 | ); | ||
18 | } | ||
19 | |||
20 | let nine = vec![9]; | ||
21 | |||
22 | let mut risk_level: u32 = 0; | ||
23 | |||
24 | for (idx, height) in points.iter().enumerate() { | ||
25 | for (ndx, num) in height.iter().enumerate() { | ||
26 | let up = points | ||
27 | .get(((idx as isize) - 1) as usize) | ||
28 | .unwrap_or_else(|| &nine) | ||
29 | .get(ndx) | ||
30 | .unwrap_or_else(|| &9); | ||
31 | |||
32 | let down = match points.get(idx + 1) { | ||
33 | Some(line) => match line.get(ndx) { | ||
34 | Some(number) => number, | ||
35 | None => &9, | ||
36 | }, | ||
37 | None => &9, | ||
38 | }; | ||
39 | let left = match height.get((ndx as isize - 1) as usize) { | ||
40 | Some(number) => number, | ||
41 | None => &9, | ||
42 | }; | ||
43 | let right = match height.get(ndx + 1) { | ||
44 | Some(number) => number, | ||
45 | None => &9, | ||
46 | }; | ||
47 | if num < up && num < down && num < right && num < left { | ||
48 | risk_level += 1 + *num; | ||
49 | } | ||
50 | } | ||
51 | } | ||
52 | println!("{}", risk_level); | ||
53 | } | ||
54 | |||
55 | fn open_file() -> BufReader<File> { | ||
56 | let args: Vec<String> = env::args().collect(); | ||
57 | |||
58 | if args.len() != 2 { | ||
59 | eprintln!("Usage: {} <filename>", args[0]); | ||
60 | std::process::exit(1); | ||
61 | } | ||
62 | |||
63 | let filename = &args[1]; | ||
64 | |||
65 | let file = File::open(filename).unwrap_or_else(|_| panic!("No such file: {}", filename)); | ||
66 | BufReader::new(file) | ||
67 | } | ||