diff options
| -rw-r--r-- | challenge-151/lubos-kolouch/python/ch-1.py | 30 |
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 |