feat: completed day 6

ankjevel · ankjevel · commit 5f2686f259b0 · 2020-12-06T10:04:42.000+01:00
diff --git a/README.md b/README.md
@@ -8,7 +8,7 @@
 - [Day 3](https://github.com/ankjevel/adventofcode/tree/2020/day_03) 🌟 🌟
 - [Day 4](https://github.com/ankjevel/adventofcode/tree/2020/day_04) 🌟 🌟
 - [Day 5](https://github.com/ankjevel/adventofcode/tree/2020/day_05) 🌟 🌟
-- [Day 6](#)
+- [Day 6](https://github.com/ankjevel/adventofcode/tree/2020/day_06) 🌟 🌟
 - [Day 7](#)
 - [Day 8](#)
 - [Day 9](#)
diff --git a/day_06/Cargo.toml b/day_06/Cargo.toml
@@ -0,0 +1,9 @@
+[package]
+name = "day_06"
+version = "0.1.0"
+authors = ["Dennis Pettersson <mail@dennispettersson.se>"]
+edition = "2018"
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
diff --git a/day_06/src/lib.rs b/day_06/src/lib.rs
@@ -0,0 +1,45 @@
+pub mod part_01;
+pub mod part_02;
+
+pub type Input = Vec<Vec<String>>;
+
+pub fn parse_input(input: &str) -> Input {
+    input
+        .lines()
+        .map(|string| string.trim())
+        .into_iter()
+        .fold(vec![vec![]], |mut list, string| {
+            if string.is_empty() {
+                list.push(vec![]);
+            } else {
+                let last = list.last_mut().unwrap();
+                last.push(string.to_string());
+            }
+            list
+        })
+        .into_iter()
+        .filter(|list| list.len() > 0)
+        .collect()
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        abc
+
+        a
+        b
+
+
+        ";
+
+    #[test]
+    fn it_parses_example() {
+        assert_eq!(
+            parse_input(&EXAMPLE_DATA),
+            vec![vec!["abc"], vec!["a", "b"]]
+        );
+    }
+}
diff --git a/day_06/src/main.rs b/day_06/src/main.rs
@@ -0,0 +1,12 @@
+use std::io::Result;
+
+use day_06::{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_06"));
+
+    println!("part_01: {:?}", part_01(&input).unwrap());
+    println!("part_02: {:?}", part_02(&input).unwrap());
+
+    Ok(())
+}
diff --git a/day_06/src/part_01.rs b/day_06/src/part_01.rs
@@ -0,0 +1,50 @@
+use std::io::Result;
+
+use crate::Input;
+
+pub fn main(input: &Input) -> Result<usize> {
+    Ok(input.iter().fold(0, |sum, group| {
+        sum + ('a'..='z')
+            .map(char::from)
+            .filter(|c| {
+                group
+                    .iter()
+                    .filter(|p| p.contains(&c.to_string()))
+                    .collect::<Vec<_>>()
+                    .len()
+                    > 0
+            })
+            .collect::<Vec<char>>()
+            .len()
+    }))
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::parse_input;
+
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        abc
+
+        a
+        b
+        c
+        
+        ab
+        ac
+        
+        a
+        a
+        a
+        a
+        
+        b
+    ";
+
+    #[test]
+    fn it_gets_the_example_correct() {
+        assert_eq!(main(&parse_input(&EXAMPLE_DATA)).unwrap(), 11);
+    }
+}
diff --git a/day_06/src/part_02.rs b/day_06/src/part_02.rs
@@ -0,0 +1,50 @@
+use std::io::Result;
+
+use crate::Input;
+
+pub fn main(input: &Input) -> Result<usize> {
+    Ok(input.iter().fold(0, |sum, group| {
+        sum + ('a'..='z')
+            .map(char::from)
+            .filter(|c| {
+                group
+                    .iter()
+                    .filter(|p| p.contains(&c.to_string()))
+                    .collect::<Vec<_>>()
+                    .len()
+                    == group.len()
+            })
+            .collect::<Vec<char>>()
+            .len()
+    }))
+}
+
+#[cfg(test)]
+mod tests {
+    use crate::parse_input;
+
+    use super::*;
+
+    const EXAMPLE_DATA: &'static str = "
+        abc
+
+        a
+        b
+        c
+        
+        ab
+        ac
+        
+        a
+        a
+        a
+        a
+        
+        b
+    ";
+
+    #[test]
+    fn it_gets_the_example_correct() {
+        assert_eq!(main(&parse_input(&EXAMPLE_DATA)).unwrap(), 6);
+    }
+}
diff --git a/input/day_06 b/input/day_06
