import 2020 aoc

2020/day6/day6.py

@@ -0,0 +1,29 @@
+#! /usr/bin/env python3
+from collections import Counter
+
+
+def part1(inp):
+    groups = open(inp).read().split("\n\n")
+    number_of_questions = 0
+    for group in groups:
+        unique_questions = set(group.replace("\n", ""))
+        number_of_questions += len(unique_questions)
+    print(number_of_questions)
+
+
+def part2(inp):
+    # number of questions for which everyone in a group answered 'yes'
+    number_of_questions = 0
+    groups = open(inp).read().split("\n\n")
+
+    for group in groups:
+        group_length = group.count("\n") + 1
+        group_counter = Counter(group.replace("\n", ""))
+        everyone_answered = [k for (k, v) in group_counter.items() if v == group_length]
+        number_of_questions += len(everyone_answered)
+    print(number_of_questions)
+
+
+if __name__ == "__main__":
+    part1("input.txt")
+    part2("input.txt")