aocpy/2015/d1.py

23 lines
318 B
Python
Raw Normal View History

2024-01-28 04:56:28 +01:00
from lib import *
data = open(0).read()
part_2 = True
floor = 0
for i, c in enumerate(data):
if c == "(":
floor += 1
elif c == ")":
floor -= 1
else:
print(c)
assert False
if part_2 and floor == -1:
print(i + 1)
break
if not part_2:
print(floor)