From f31334b43862ed7dfd35551db45ef416c97810be Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Sun, 6 Dec 2020 07:23:29 +0100
Subject: [PATCH] 2020: d06: ex2: add solution

---
 2020/d06/ex2/ex2.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100755 2020/d06/ex2/ex2.py

diff --git a/2020/d06/ex2/ex2.py b/2020/d06/ex2/ex2.py
new file mode 100755
index 0000000..c78283d
--- /dev/null
+++ b/2020/d06/ex2/ex2.py
@@ -0,0 +1,31 @@
+#!/usr/bin/env python
+
+import sys
+from collections import defaultdict
+from typing import DefaultDict, List, Set
+
+
+def solve(raw: List[str]) -> int:
+    answers: DefaultDict[int, Set[str]] = defaultdict(set)
+    group = 0
+    first = True
+    for line in raw:
+        if line == "":
+            group += 1
+            first = True
+            continue
+        if not first:
+            answers[group] &= {char for char in line}
+        else:
+            answers[group] = {char for char in line}
+        first = False
+    return sum(len(answer) for answer in answers.values())
+
+
+def main() -> None:
+    input = [line.strip() for line in sys.stdin.readlines()]
+    print(solve(input))
+
+
+if __name__ == "__main__":
+    main()