29 lines
522 B
Python
29 lines
522 B
Python
# 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}")
|