From 8a285dcf05026514241528dcf9c7b2f7adb17592 Mon Sep 17 00:00:00 2001 From: dece Date: Sun, 6 Dec 2020 16:59:28 +0100 Subject: [PATCH] Day 6 --- 2020/day6.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 2020/day6.py diff --git a/2020/day6.py b/2020/day6.py new file mode 100644 index 0000000..cee3493 --- /dev/null +++ b/2020/day6.py @@ -0,0 +1,27 @@ +def main(): + with open("day6.txt", "rt") as f: + text = f.read() + groups = text.split("\n\n") + + # Part 1 + count = 0 + for g in groups: + letters = set(g) + count += len(letters) + if "\n" in g: + count -= 1 + print("Total:", count) + + # Part 2 + count = 0 + for g in groups: + common = None + answers = [set(p) for p in g.split("\n") if p] + for a in answers: + common = set(a) if common is None else common & set(a) + count += len(common) + print("Commons:", count) + + +if __name__ == "__main__": + main()