day 22 cleanup

main
magical 2023-12-22 08:54:07 +00:00
parent efd7f887b8
commit 3dff6a2247
1 changed files with 22 additions and 39 deletions

View File

@ -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)