feat(day_07): somewhat sloppy solution

ankjevel · ankjevel · commit f0c007c7d93a · 2022-12-07T11:04:52.000+01:00
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@
 - [Day 4](https://github.com/ankjevel/adventofcode/tree/2022/day_04) ⭐️ ⭐️
 - [Day 5](https://github.com/ankjevel/adventofcode/tree/2022/day_05) ⭐️ ⭐️
 - [Day 6](https://github.com/ankjevel/adventofcode/tree/2022/day_06) ⭐️ ⭐️
-- [Day 7](#)
+- [Day 7](https://github.com/ankjevel/adventofcode/tree/2022/day_07) ⭐️ ⭐️
 - [Day 8](#)
 - [Day 9](#)
 - [Day 10](#)
diff --git a/day_07/Cargo.toml b/day_07/Cargo.toml
@@ -0,0 +1,15 @@
+[package]
+name = "day_07"
+version = "0.1.0"
+edition = "2021"
+authors = ["Dennis Pettersson <mail@dennispettersson.se>"]
+
+[lib]
+doctest = false
+
+[[bin]]
+name = "day_07"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
diff --git a/day_07/src/lib.rs b/day_07/src/lib.rs
@@ -0,0 +1,27 @@
+pub mod part_01;
+pub mod part_02;
+
+pub type Input = Vec<String>;
+
+pub fn parse_input(input: &str) -> Input {
+    input
+        .lines()
+        .map(str::trim)
+        .filter(|string| !string.is_empty())
+        .map(str::to_owned)
+        .collect()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        example
+    ";
+
+    #[test]
+    fn it_parses_example() {
+        assert_eq!(parse_input(&EXAMPLE_DATA), vec!["example"]);
+    }
+}
diff --git a/day_07/src/main.rs b/day_07/src/main.rs
@@ -0,0 +1,12 @@
+use std::io::Result;
+
+use day_07::{parse_input, part_01::main as part_01, part_02::main as part_02};
+
+fn main() -> Result<()> {
+    let input = parse_input(include_str!("../../input/day_07"));
+
+    println!("part_01: {:?}", part_01(&input)?);
+    println!("part_02: {:?}", part_02(&input)?);
+
+    Ok(())
+}
diff --git a/day_07/src/part_01.rs b/day_07/src/part_01.rs
@@ -0,0 +1,83 @@
+use std::{
+    collections::{HashMap, VecDeque},
+    io::Result,
+};
+
+use crate::Input;
+
+pub fn main(input: &Input) -> Result<u32> {
+    let mut path: VecDeque<String> = VecDeque::new();
+    let mut dirs: HashMap<String, u32> = HashMap::new();
+
+    for row in input.into_iter() {
+        if row.starts_with('$') {
+            if row.contains("$ cd") {
+                let next_dir = row.replace("$ cd ", "");
+                if next_dir.contains("..") {
+                    path.pop_back();
+                } else {
+                    path.push_back(if next_dir.contains('/') {
+                        next_dir
+                    } else {
+                        format!("{}/", next_dir)
+                    });
+                }
+            }
+        } else {
+            if !row.starts_with("dir ") {
+                let n: Vec<&str> = row.split(' ').collect();
+                let size = n[0].parse::<u32>().unwrap();
+                let mut current: Vec<String> = vec![];
+                for path in path.clone().into_iter() {
+                    current.push(path);
+                    let path = current.join("");
+                    *dirs.entry(path).or_insert(0) += size;
+                }
+            }
+        }
+    }
+
+    Ok(dirs
+        .into_values()
+        .filter(|val| val <= &100_000)
+        .sum::<u32>())
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::parse_input;
+
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        $ cd /
+        $ ls
+        dir a
+        14848514 b.txt
+        8504156 c.dat
+        dir d
+        $ cd a
+        $ ls
+        dir e
+        29116 f
+        2557 g
+        62596 h.lst
+        $ cd e
+        $ ls
+        584 i
+        $ cd ..
+        $ cd ..
+        $ cd d
+        $ ls
+        4060174 j
+        8033020 d.log
+        5626152 d.ext
+        7214296 k
+    ";
+
+    #[test]
+    fn it_gets_the_example_correct() -> Result<()> {
+        assert_eq!(main(&parse_input(&EXAMPLE_DATA))?, 95437);
+        Ok(())
+    }
+}
diff --git a/day_07/src/part_02.rs b/day_07/src/part_02.rs
@@ -0,0 +1,87 @@
+use std::{
+    collections::{HashMap, VecDeque},
+    io::Result,
+};
+
+use crate::Input;
+
+pub fn main(input: &Input) -> Result<u32> {
+    let mut path: VecDeque<String> = VecDeque::new();
+    let mut dirs: HashMap<String, u32> = HashMap::new();
+
+    for row in input.into_iter() {
+        if row.starts_with('$') {
+            if row.contains("$ cd") {
+                let next_dir = row.replace("$ cd ", "");
+                if next_dir.contains("..") {
+                    path.pop_back();
+                } else {
+                    path.push_back(if next_dir.contains('/') {
+                        next_dir
+                    } else {
+                        format!("{}/", next_dir)
+                    });
+                }
+            }
+        } else {
+            if !row.starts_with("dir ") {
+                let n: Vec<&str> = row.split(' ').collect();
+                let size = n[0].parse::<u32>().unwrap();
+                let mut current: Vec<String> = vec![];
+                for path in path.clone().into_iter() {
+                    current.push(path);
+                    let path = current.join("");
+                    *dirs.entry(path).or_insert(0) += size;
+                }
+            }
+        }
+    }
+
+    let to_remove = 30_000_000 - (70_000_000 - dirs.get("/").unwrap_or(&0));
+
+    Ok(dirs
+        .clone()
+        .into_values()
+        .filter(|val| val >= &to_remove)
+        .min()
+        .unwrap())
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::parse_input;
+
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        $ cd /
+        $ ls
+        dir a
+        14848514 b.txt
+        8504156 c.dat
+        dir d
+        $ cd a
+        $ ls
+        dir e
+        29116 f
+        2557 g
+        62596 h.lst
+        $ cd e
+        $ ls
+        584 i
+        $ cd ..
+        $ cd ..
+        $ cd d
+        $ ls
+        4060174 j
+        8033020 d.log
+        5626152 d.ext
+        7214296 k
+    ";
+
+    #[test]
+    fn it_gets_the_example_correct() -> Result<()> {
+        assert_eq!(main(&parse_input(&EXAMPLE_DATA))?, 24933642);
+        Ok(())
+    }
+}
diff --git a/input/day_07 b/input/day_07
