day1.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17

#!/usr/bin/env python

with open('day1.txt') as data:
    floors = data.read().strip()
# part 1
print(sum(map(lambda x: 1 if x == '(' else -1, floors)))

# part 2
floor = 0
for i, char in enumerate(floors):
    if char == '(':
        floor += 1
    else:
        floor -= 1
    if floor == -1:
        print(i+1)
        break