Day 3: Gear Ratios
Megathread guidelines
- Keep top level comments as only solutions, if you want to say something other than a solution put it in a new post. (replies to comments can be whatever)
- Code block support is not fully rolled out yet but likely will be in the middle of the event. Try to share solutions as both code blocks and using something such as https://topaz.github.io/paste/ or pastebin (code blocks to future proof it for when 0.19 comes out and since code blocks currently function in some apps and some instances as well if they are running a 0.19 beta)
FAQ
- What is this?: Here is a post with a large amount of details: https://programming.dev/post/6637268
- Where do I participate?: https://adventofcode.com/
- Is there a leaderboard for the community?: We have a programming.dev leaderboard with the info on how to join in this post: https://programming.dev/post/6631465
🔒This post will be unlocked when there is a decent amount of submissions on the leaderboard to avoid cheating for top spots
🔓 Edit: Post has been unlocked after 11 minutes
Language: Python
Classic AoC grid problem… Tedious as usual, but very doable. Took my time and I’m pretty happy with the result. :]
Part 1
For the first part, I decided to break the problem into: 1. Reading the schematic, 2. Finding the numbers, 3. Finding the parts. This was useful for Part 2 as I could re-use my
read_schematic
andfind_numbers
functions.Two things I typically do for grid problems:
DIRECTIONS
list I loop through so I can check easily check the neighbors.Schematic = list[str] Number = tuple[int, int, int] DIRECTIONS = ( (-1, -1), (-1, 0), (-1, 1), ( 0, -1), ( 0, 1), ( 1, -1), ( 1, 0), ( 1, 1), ) def read_schematic(stream=sys.stdin) -> Schematic: schematic = [line.strip() for line in stream] columns = len(schematic[0]) + 2 return [ '.'*columns, *['.' + line + '.' for line in schematic], '.'*columns, ] def is_symbol(s: str) -> bool: return not (s.isdigit() or s == '.') def find_numbers(schematic: Schematic) -> Iterator[Number]: rows = len(schematic) columns = len(schematic[0]) for r in range(1, rows): for number in re.finditer(r'[0-9]+', schematic[r]): yield (r, *number.span()) def find_parts(schematic: Schematic, numbers: Iterator[Number]) -> Iterator[int]: for r, c_head, c_tail in numbers: part = int(schematic[r][c_head:c_tail]) for c in range(c_head, c_tail): neighbors = (schematic[r + dr][c + dc] for dr, dc in DIRECTIONS) if any(is_symbol(neighbor) for neighbor in neighbors): yield part break def main(stream=sys.stdin) -> None: schematic = read_schematic(stream) numbers = find_numbers(schematic) parts = find_parts(schematic, numbers) print(sum(parts))
Part 2
For the second part, I just found the stars, and then I found the gears by checking if the stars are next to two numbers (which I had found previously).
def find_stars(schematic: Schematic) -> Iterator[Star]: rows = len(schematic) columns = len(schematic[0]) for r in range(1, rows): for c in range(1, columns): token = schematic[r][c] if token == '*': yield (r, c) def find_gears(schematic: Schematic, stars: Iterator[Star], numbers: list[Number]) -> Iterator[int]: for star_r, star_c in stars: gears = [ int(schematic[number_r][number_c_head:number_c_tail]) for number_r, number_c_head, number_c_tail in numbers if any(star_r + dr == number_r and number_c_head <= (star_c + dc) < number_c_tail for dr, dc in DIRECTIONS) ] if len(gears) == 2: yield gears[0] * gears[1] def main(stream=sys.stdin) -> None: schematic = read_schematic(stream) numbers = find_numbers(schematic) stars = find_stars(schematic) gears = find_gears(schematic, stars, list(numbers)) print(sum(gears))
GitHub Repo