summaryrefslogtreecommitdiff
path: root/01/main.py
diff options
context:
space:
mode:
Diffstat (limited to '01/main.py')
-rwxr-xr-x01/main.py35
1 files changed, 35 insertions, 0 deletions
diff --git a/01/main.py b/01/main.py
new file mode 100755
index 0000000..cd3ec34
--- /dev/null
+++ b/01/main.py
@@ -0,0 +1,35 @@
+#!/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()