From b4a3bd52e5e610f431d9aa49e3ac0af918e1e344 Mon Sep 17 00:00:00 2001
From: Cody Hiar <cody@hiar.ca>
Date: Fri, 5 Feb 2021 09:29:54 -0700
Subject: Adding day 5 and 6

---
 day6/main.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)
 create mode 100644 day6/main.py

(limited to 'day6/main.py')

diff --git a/day6/main.py b/day6/main.py
new file mode 100644
index 0000000..7bad5d2
--- /dev/null
+++ b/day6/main.py
@@ -0,0 +1,24 @@
+"""Day 6."""
+from collections import Counter
+
+
+def score_part1(group):
+    """Determine score for group in part 1."""
+    return len(set(group.replace("\n", "")))
+
+
+def score_part2(group):
+    """Determine score for group in part 2."""
+    group_size = group.count("\n") + 1
+    counter = Counter(group.replace("\n", ""))
+    return sum(1 for _, occurences in counter.items() if occurences == group_size)
+
+
+with open("input") as f:
+    groups = f.read().rstrip().split("\n\n")
+
+# Part 1
+print(sum(map(score_part1, groups)))
+
+# Part 2
+print(sum(map(score_part2, groups)))
-- 
cgit v1.2.3