Challenge 151 LK Task 1 Python

1 files changed, 30 insertions, 0 deletions

diff --git a/challenge-151/lubos-kolouch/python/ch-1.py b/challenge-151/lubos-kolouch/python/ch-1.py
new file mode 100644
index 0000000000..fd34a05784
--- /dev/null
+++ b/challenge-151/lubos-kolouch/python/ch-1.py
@@ -0,0 +1,30 @@
+import re
+
+
+def get_min_depth(input: str):
+    """Calculate the depth"""
+    # Input: '1 | 2 3 | 4 5'
+
+    # iterate through the layers. If the next layer does not have 2^n items,
+    # there must be a leaf node
+
+    input = re.sub(r"\s", "", input)
+    layers = input.split("|")
+
+    for layer_count, _ in enumerate(layers, 1):
+        # if not defined means we are at the last layer
+
+        try:
+            items_count = len(layers[layer_count])
+        except IndexError:
+            return layer_count
+
+        if items_count != 2**layer_count:
+            return layer_count
+
+    return None
+
+
+assert get_min_depth("1 | 2 3 | 4 5") == 2
+assert get_min_depth("1 | 2 3 | 4 *  * 5 | * 6") == 3
+assert get_min_depth("1 | 2 3") == 2