62 lines
1.3 KiB
Python
62 lines
1.3 KiB
Python
import itertools
|
|
rocks = [
|
|
0b00011110,
|
|
0b00001000_00011100_00001000,
|
|
0b00000100_00000100_00011100,
|
|
0b00010000_00010000_00010000_00010000,
|
|
0b00011000_00011000,
|
|
]
|
|
|
|
jets = itertools.cycle(">>><<><>><<<>><>>><<<>>><<<><<<>><>><<>>")
|
|
jets = itertools.cycle(open("input").read().strip())
|
|
|
|
rocki = itertools.cycle(rocks)
|
|
|
|
a = 0b11111111
|
|
wall = 0b10000000_10000000_10000000_10000000_10000000
|
|
stopped = 0
|
|
r = 0
|
|
|
|
def drop(a, r, jets):
|
|
# start falling
|
|
h = a.bit_length() + 8*3
|
|
if h % 8 != 0:
|
|
h += 8 - h%8
|
|
while True:
|
|
j = next(jets)
|
|
if j == '<':
|
|
x = (r<<1)
|
|
if x & wall == 0:
|
|
if x & (a>>h) == 0:
|
|
r = x
|
|
elif j == '>':
|
|
x = (r<<7)
|
|
if x & wall == 0:
|
|
x >>= 8
|
|
if x & (a>>h) == 0:
|
|
r = x
|
|
|
|
# continue falling?
|
|
if r & (a>>(h-8)) == 0:
|
|
h -= 8
|
|
assert h > 0
|
|
else:
|
|
# stop
|
|
a |= (r<<h)
|
|
return a
|
|
|
|
|
|
a = 0b11111111
|
|
wall = 0b10000000_10000000_10000000_10000000_10000000
|
|
stopped = 0
|
|
while stopped < 2022:
|
|
r = next(rocki)
|
|
a = drop(a,r,jets)
|
|
stopped += 1
|
|
#for i in reversed(range(0,a.bit_length(),8)):
|
|
# print("{:08b}".format((a>>i)&0xff))
|
|
#print()
|
|
|
|
import math
|
|
print(math.ceil((a>>8).bit_length()/8))
|