[py] Port 2015_01

py/aoc/y2015/d01.py

@@ -0,0 +1,29 @@
+# PART 1
+
+
+def count_floors(steps):
+    return steps.count("(") - steps.count(")")
+
+
+# PART 2
+
+
+def find_basement_char(steps):
+    pos = 0
+    step_nr = 0
+    for step in steps:
+        step_nr += 1
+        if step == "(":
+            pos += 1
+        if step == ")":
+            pos -= 1
+        if pos < 0:
+            return step_nr
+
+
+def solve(inputstr):
+    steps = inputstr.strip()
+    floor = count_floors(steps)
+    print(f"Part 1: {floor}")
+    step_nr = find_basement_char(steps)
+    print(f"Part 2: {step_nr}")