day 22 cleanup

day22/sol.py

@@ -38,7 +38,7 @@ def solve(f):
     check(data)
     settle(data)
     check(data)
-    #disintegrate(data)
+    disintegrate(data)
     cascade(data)
 
 def settle(data):
@@ -46,26 +46,28 @@ def settle(data):
         return b[2]
     data.sort(key=zindex)
 
-  
     top = 1
     for i in range(len(data)):
-        # first, lower all blocks >i so that at least one block is at z=top
-        minz = min(z[0] for x,y,z in data[i:])
-        if minz > top:
-            print("lowering all blocks by", minz-top)
-            for b in data[i:]:
-                b[:] = lower(b, minz-top)
-
-        # lower blocks one at a time
+        # first, lower to just above the top of the highest block
         b = data[i]
+        z = b[2][0]
+        if z > top:
+            #print("lowering block %d by %d" % (i, z-top))
+            b = lower(b, z-top)
+
+        # lower block one at a time
         n = 0
-        while canlower(b, data[:i]):
+        under = data[:i]
+        while canlower(b, under):
             b = lower(b)
             n += 1
-        print("lowering block %d by %d" % (i,n))
+        print("lowering block %d by %d+%d" % (i,z-top,n))
         data[i] = b
 
-        top = max(z[1] for x,y,z in data[:i+1]) + 1
+        top = max(top, b[2][1] + 1)
+
+    # FIXME: should probably re-sort the data here,
+    #        but it seems to work even if we don't
 
 
 def canlower(block, under):
@@ -86,10 +88,8 @@ def disintegrate(data):
 
     for i,x in enumerate(data):
         # disintegrate x
-        # find other blocks on the same z
-        #others = [y for y in data if x != y and not (y[2][1] < x[2][0] or y[2][0] > x[2][1])]
 
-        # see if any block can be lowered
+        # see if any block above x can be lowered
         # if yes, then x cannot be disintegrated
         candisintegrate = True
         for j in range(i+1, len(data)):
@@ -97,24 +97,10 @@ def disintegrate(data):
                 candisintegrate = False
                 break
 
-            
-            #ly = lower(y)
-            #if y == ly:
-            #    continue
-            #supports = [z for z in others if z != x and z != y and overlaps(z,ly)]
-            #if supports:
-            #    # theer is a block that supports y
-            #    #candisintegrate = True
-            #    print('y = %s would be blocked by %d blocks' % (y, len(supports)))
-            #    pass
-            #else:
-            #    # no block would stop y from falling
-            #    print("no blockk stops y=%s" % (y))
-            #    candisintegrate = False
-            #    break
         print(i, x, candisintegrate)
         if candisintegrate:
             t += 1
+
     print(t)
     return (t)
 
@@ -160,12 +146,9 @@ def cascade(data):
 def lower(brick, n=1):
     """lower a brick n spaces in the z direction"""
     x, y, z = brick
-    if z[0] > 1:
-        z = z[0]-1, z[1]-1
+    if z[0] > n:
+        z = z[0]-n, z[1]-n
     return [x, y, z]
 
 import sys
 solve(sys.stdin)
-
-
-