day 14 python cleanup

day14/sol.py

@@ -9,45 +9,32 @@ for line in open("input"):
     for point in line.replace("->", " ").split():
         x, y = map(int, point.split(","))
         path.append((x,y))
-        xs.append(x)
-        ys.append(y)
     data.append(path)
 
 
-grid = []
-sand = {}
+grid = {}
 
-X0 = min(xs)
-X1 = max(xs) + 1
+X0 = 500
+X1 = 500 + 1
 Y0 = 0
-Y1 = max(ys) + 1
-
-# part 2:
-X0 -= 1000
-X1 += 1000
-Y1 += 1
-
-DX = X1 - X0
-DY = Y1 - Y0
-
-grid = [['.']*DX for _ in range(DY)]
+Y1 = 1
 
 def put(x,y,c):
-    if c == 'o':
-        sand[(x,y)] = c
-    if X0 <= x < X1:
-        if Y0 <= y < Y1:
-            grid[y-Y0][x-X0] = c
-            return
-    print(x,y,"out of range")
+    global X0, X1, Y0, Y1
+    if x < X0:
+        X0 = x
+    if x >= X1:
+        X1 = x + 1
+    if y < Y0:
+        Y0 = y
+    if y >= Y1:
+        Y1 = y + 1
+    grid[(x,y)] = c
 
 def get(x,y):
-    if (x,y) in sand:
-        return 'o'
-    if X0 <= x < X1:
-        if Y0 <= y < Y1:
-            return grid[y-Y0][x-X0]
-    return '#'
+    if (x,y) in grid:
+        return grid[(x,y)]
+    return '.'
 
 def draw(path):
     for p1, p2 in zip(path, path[1:]):
@@ -73,30 +60,31 @@ for path in data:
 
 
 def show():
-    for r in grid: print("".join(r))
+    for y in range(Y0, Y1):
+        print("".join(get(x,y) for x in range(X0, X1)))
 
-#show()
+show()
 
-def drip(p):
+def drip(p, floor=None):
     x,y=p
     while True:
-        c = get(x,y+1)
-        if c == '?':
+        if y+1 == floor:
+            # blocked
+            put(x,y,'o')
+            return 'done'
+        if y >= Y1:
             return 'bye'
+        c = get(x,y+1)
         if c == '.':
             y += 1
             continue
         # try left
         c = get(x-1,y+1)
-        if c == '?':
-            return 'bye'
         if c == '.':
             x,y = x-1,y+1
             continue
         # try right
         c = get(x+1, y+1)
-        if c == '?':
-            return 'bye'
         if c == '.':
             x,y = x+1, y+1
             continue
@@ -105,16 +93,24 @@ def drip(p):
         #print(x,y)
         return 'done'
 
-#i = 0
-#for _ in range(100000):
-#    if drip(source) == 'bye':
-#        break
-#    i+= 1
-
 i = 0
-while source not in sand:
-    drip(source)
-    i += 1
-
-#show()
+while True:
+    if drip(source) == 'bye':
+        break
+    i+= 1
+print(i)
+
+floor = Y1 + 1
+Y1 = floor+1
+show()
+
+while source not in grid:
+    r = drip(source, floor=floor)
+    i += 1
+    if r == 'bye':
+        break
+
+# 28594
+
+show()
 print(i)