2022 Day 15 Python

This commit is contained in:
Luke Hubmayer-Werner 2022-12-15 16:31:56 +10:30
parent b56b19720b
commit 0e3f7c9bc4
2 changed files with 92 additions and 0 deletions

64
2022/day15.py Normal file
View File

@ -0,0 +1,64 @@
from helpers import *
lines = read_day(15).split('\n')
sample_lines = '''
Sensor at x=2, y=18: closest beacon is at x=-2, y=15
Sensor at x=9, y=16: closest beacon is at x=10, y=16
Sensor at x=13, y=2: closest beacon is at x=15, y=3
Sensor at x=12, y=14: closest beacon is at x=10, y=16
Sensor at x=10, y=20: closest beacon is at x=10, y=16
Sensor at x=14, y=17: closest beacon is at x=10, y=16
Sensor at x=8, y=7: closest beacon is at x=2, y=10
Sensor at x=2, y=0: closest beacon is at x=2, y=10
Sensor at x=0, y=11: closest beacon is at x=2, y=10
Sensor at x=20, y=14: closest beacon is at x=25, y=17
Sensor at x=17, y=20: closest beacon is at x=21, y=22
Sensor at x=16, y=7: closest beacon is at x=15, y=3
Sensor at x=14, y=3: closest beacon is at x=15, y=3
Sensor at x=20, y=1: closest beacon is at x=15, y=3
'''.strip().split('\n')
def do_sensors(lines: list[str], row: int):
seen_cells_on_row = set()
for line in lines:
x, y, nx, ny = line_to_numbers(line)
distance = abs(x-nx) + abs(y-ny)
dr = abs(y-row)
if dr < distance:
d2 = distance - dr
seen_cells_on_row |= {x+i for i in range(-d2, d2+1)}
print(f'Part 1: {len(seen_cells_on_row) - 1}')
# do_sensors(sample_lines, 20)
do_sensors(lines, 2_000_000)
def in_range_of_sensors(sensors, x, y):
for (i,j,d) in sensors:
if abs(x-i) + abs(y-j) <= d:
d2 = d - abs(x-i)
y_advance = (j+d2)-y + 1
return y_advance
return False
def do_sensors_2(lines: list[str], search_space: int):
sensors = []
for num, line in enumerate(lines, 1):
x, y, nx, ny = line_to_numbers(line)
if not ((0 <= x <= 4_000_000) and (0 <= y <= 4_000_000)):
continue
distance = abs(x-nx) + abs(y-ny)
sensors.append((x, y, distance))
# print(sensors)
for x in range(0, search_space+1):
# if x % 100_000 == 0:
# print(f'Processed {x} rows')
y = 0
while y <= search_space:
y_skip = in_range_of_sensors(sensors, x, y)
if y_skip is False:
print(f'Part 2: Seen at ({x},{y}) = submit {x*4000000 + y}')
return
y += y_skip
# do_sensors_2(sample_lines, 20)
do_sensors_2(lines, 4_000_000)

28
2022/input/15 Normal file
View File

@ -0,0 +1,28 @@
Sensor at x=193758, y=2220950: closest beacon is at x=652350, y=2000000
Sensor at x=3395706, y=3633894: closest beacon is at x=3404471, y=3632467
Sensor at x=3896022, y=3773818: closest beacon is at x=3404471, y=3632467
Sensor at x=1442554, y=1608100: closest beacon is at x=652350, y=2000000
Sensor at x=803094, y=813675: closest beacon is at x=571163, y=397470
Sensor at x=3491072, y=3408908: closest beacon is at x=3404471, y=3632467
Sensor at x=1405010, y=486446: closest beacon is at x=571163, y=397470
Sensor at x=3369963, y=3641076: closest beacon is at x=3404471, y=3632467
Sensor at x=3778742, y=2914974: closest beacon is at x=4229371, y=3237483
Sensor at x=1024246, y=3626229: closest beacon is at x=2645627, y=3363491
Sensor at x=3937091, y=2143160: closest beacon is at x=4229371, y=3237483
Sensor at x=2546325, y=2012887: closest beacon is at x=2645627, y=3363491
Sensor at x=3505386, y=3962087: closest beacon is at x=3404471, y=3632467
Sensor at x=819467, y=239010: closest beacon is at x=571163, y=397470
Sensor at x=2650614, y=595151: closest beacon is at x=3367919, y=-1258
Sensor at x=3502942, y=6438: closest beacon is at x=3367919, y=-1258
Sensor at x=3924022, y=634379: closest beacon is at x=3367919, y=-1258
Sensor at x=2935977, y=2838245: closest beacon is at x=2645627, y=3363491
Sensor at x=1897626, y=7510: closest beacon is at x=3367919, y=-1258
Sensor at x=151527, y=640680: closest beacon is at x=571163, y=397470
Sensor at x=433246, y=1337298: closest beacon is at x=652350, y=2000000
Sensor at x=2852855, y=3976978: closest beacon is at x=3282750, y=3686146
Sensor at x=3328398, y=3645875: closest beacon is at x=3282750, y=3686146
Sensor at x=3138934, y=3439134: closest beacon is at x=3282750, y=3686146
Sensor at x=178, y=2765639: closest beacon is at x=652350, y=2000000
Sensor at x=3386231, y=3635056: closest beacon is at x=3404471, y=3632467
Sensor at x=3328074, y=1273456: closest beacon is at x=3367919, y=-1258
Sensor at x=268657, y=162438: closest beacon is at x=571163, y=397470