#!/usr/bin/python def solve_partn(partn, data): floor = 0 for i, d in enumerate(data): if d == "(": floor += 1 elif d == ")": floor -= 1 if partn == 2: if floor < 0: return i + 1 if partn == 1: return floor else: # partn == 2 return -1 # only reached if he never goes underground def main(): # Read instructions from input text file with open("input.txt", "r") as f: data = f.read() print("Part 1 solution:", solve_partn(1, data)) # 74 for me print("Part 2 solution:", solve_partn(2, data)) # 1795 for me if __name__ == "__main__": main()