From 026ec029b34da00b3754665b98fa2fff6f63cf08 Mon Sep 17 00:00:00 2001
From: Bruno BELANYI <bruno@belanyi.fr>
Date: Sun, 8 Dec 2019 22:13:54 +0100
Subject: [PATCH] 2019: d06: ex1: add solution

---
 2019/d06/ex1/ex1.py | 53 +++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 53 insertions(+)
 create mode 100755 2019/d06/ex1/ex1.py

diff --git a/2019/d06/ex1/ex1.py b/2019/d06/ex1/ex1.py
new file mode 100755
index 0000000..d4b3cf9
--- /dev/null
+++ b/2019/d06/ex1/ex1.py
@@ -0,0 +1,53 @@
+#!/usr/bin/env python
+
+import sys
+from dataclasses import dataclass, field
+from typing import Dict, List
+
+
+@dataclass
+class OrbitGraph:
+    name: str
+    children: List["OrbitGraph"] = field(default_factory=list)
+
+
+def make_orbits(description: List[str]) -> OrbitGraph:
+    orbits: Dict[str, OrbitGraph] = {}
+
+    def find_or_add(name: str) -> OrbitGraph:
+        if name in orbits:
+            return orbits[name]
+        orbit = OrbitGraph(name)
+        orbits[name] = orbit
+        return orbit
+
+    for l in description:
+        parent, child = map(find_or_add, map(lambda x: x.strip(), l.split(")")))
+        parent.children.append(child)
+
+    return orbits["COM"]  # Assume common root is named 'COM'
+
+
+def count_orbits(root: OrbitGraph) -> int:
+    ans = 0
+    stack = 0  # Count the number of direct and indirect orbits to the current root
+
+    def helper(root: OrbitGraph):
+        nonlocal ans
+        nonlocal stack
+        ans += stack  # Count the number of orbits to this node
+        stack += 1  # Add the current root to stack of parents
+        for child in root.children:
+            helper(child)  # Count those orbits for each child
+        stack -= 1  # Remove the current root from the stack of parents
+
+    helper(root)
+    return ans
+
+
+def main() -> None:
+    print(count_orbits(make_orbits(sys.stdin.readlines())))
+
+
+if __name__ == "__main__":
+    main()