Add solution for day 17 *part 1*

2022/day17/code.py

@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+
+import sys
+from collections import defaultdict
+
+# ####
+# 
+# .#.
+# ###
+# .#.
+# 
+# ..#
+# ..#
+# ###
+# 
+# #
+# #
+# #
+# #
+# 
+# ##
+# ##
+
+ROCKS = [[['#','#','#','#']],[['.','#','.'],['#','#','#'],['.','#','.']],
+        [['#','#','#'],['.','.','#'],['.','.','#']],[['#'],['#'],['#'],['#']],[['#','#'],['#','#']]]
+
+CHAMBER = defaultdict(lambda: ['.'] * 7)
+MAX = 0
+
+def show():
+    for y in range(MAX+1,-1,-1):
+        line = '|'
+        for x in range(0,7):
+            line += CHAMBER[y][x]
+        print(line + '|')
+    print('+-------+')
+
+def fall(r, j):
+    rock = ROCKS[r % len(ROCKS)]
+    global MAX
+    x = 2
+    y = MAX + 3
+    while True:
+        jet = JETS[j]
+        if jet == '>': # push right
+            if not (x + len(rock[0])) == 7:
+                halt = False
+                for ji in range(len(rock)):
+                    for i in range(len(rock[0])):
+                        if rock[ji][i] == '#' and CHAMBER[y+ji][x+i+1] == '#': halt = True
+                if not halt: x = x + 1
+        elif jet == '<': # push left
+            halt = False
+            if not x == 0:
+                for ji in range(len(rock)):
+                    for i in range(len(rock[0])):
+                        if rock[ji][i] == '#' and CHAMBER[y+ji][x+i-1] == '#': halt = True
+                if not halt: x = x - 1
+        halt = False
+        for ji in range(len(rock)):
+            for i in range(len(rock[0])):
+                if rock[ji][i] == '#' and CHAMBER[y+ji-1][x+i] == '#':
+                    halt = True
+        if y == height or halt:
+            for b in range(y,y+len(rock)):
+                for a in range(x,x+len(rock[0])):
+                    CHAMBER[b][a] = rock[b-y][a-x] if CHAMBER[b][a] == '.' else CHAMBER[b][a]
+                if b+1 > MAX: MAX = b+1
+            break
+        j = (j + 1) % len(JETS)
+        y -= 1
+    return r+1, (j + 1) % len(JETS)
+
+if __name__ == '__main__':
+    global JETS
+    JETS = list(open(sys.argv[1]).read().strip('\n'))
+    global height
+    height = 0
+
+    # challenge 1
+    r, j = 0,0
+    for i in range(2022):
+        r,j = fall(r,j)
+    res1 = str(MAX)
+    print("challenge 1:" + "\n" +  res1 + "\n")
+
+    # challenge 2
+    # keep only last 20 values or so, check for cycles
+    for i in range(1000000000000-2022):
+        r,j = fall(r,j)
+        if CHAMBER[MAX-1] == ['#'] * 7 and r == 0:
+            print(i,MAX)
+    res2 = str(height+MAX)
+    print("challenge 2:" + "\n" +  res2 + "\n")
+

2022/day17/example

@@ -0,0 +1 @@
+>>><<><>><<<>><>>><<<>>><<<><<<>><>><<>>