Files
advent-of-code/adventofcode/2015/day1/day1.py

20 lines
479 B
Python

#!/usr/bin/env python3
def main(inp):
floor = 0
basement = -1
for i, c in enumerate(inp):
if c == "(": floor += 1
elif c == ")": floor -= 1
if basement == -1 and floor == -1:
basement = i + 1
print("Part 1: ", floor)
print("Part 2: ", basement)
if __name__ == '__main__':
import sys
infile = sys.argv[1] if len(sys.argv) > 1 else "example.txt"
with open(infile) as inp:
main(inp.read().rstrip())