day 6

1 files changed, 43 insertions, 0 deletions

diff --git a/src/2020/6/mod.rs b/src/2020/6/mod.rs
new file mode 100644
index 0000000..94dd4da
--- /dev/null
+++ b/src/2020/6/mod.rs
@@ -0,0 +1,43 @@
+pub fn part1() -> anyhow::Result<i64> {
+    let input = crate::util::read_file_str("data/6.txt")?;
+    let mut yes = std::collections::HashSet::new();
+    let mut total = 0;
+    for line in input.lines() {
+        if line == "" {
+            total += yes.len() as i64;
+            yes = std::collections::HashSet::new();
+        } else {
+            for c in line.chars() {
+                yes.insert(c);
+            }
+        }
+    }
+    total += yes.len() as i64;
+    Ok(total)
+}
+
+pub fn part2() -> anyhow::Result<i64> {
+    let input = crate::util::read_file_str("data/6.txt")?;
+    let mut yes = std::collections::HashSet::new();
+    for c in 'a'..='z' {
+        yes.insert(c);
+    }
+    let mut total = 0;
+    for line in input.lines() {
+        if line == "" {
+            total += yes.len() as i64;
+            yes = std::collections::HashSet::new();
+            for c in 'a'..='z' {
+                yes.insert(c);
+            }
+        } else {
+            for c in 'a'..='z' {
+                if !line.contains(c) {
+                    yes.remove(&c);
+                }
+            }
+        }
+    }
+    total += yes.len() as i64;
+    Ok(total)
+}