diff --git a/2020/d20/ex1/ex1.py b/2020/d20/ex1/ex1.py new file mode 100755 index 0000000..f8171f8 --- /dev/null +++ b/2020/d20/ex1/ex1.py @@ -0,0 +1,128 @@ +#!/usr/bin/env python + +import functools +import itertools +import math +import sys +from collections import defaultdict +from copy import deepcopy +from typing import Dict, Iterator, List, Set, Tuple + +Tile = List[List[str]] +Tiles = Dict[int, Tile] + + +def parse(raw: str) -> Tiles: + def parse_tile(raw: List[str]) -> Tile: + return [[line[i] for i in range(len(line))] for line in raw] + + res = {} + for lines in map(lambda s: s.splitlines(), raw.split("\n\n")): + if len(lines) == 0: + continue + nums = int(lines[0][5:-1]) + res[nums] = parse_tile(lines[1:]) + return res + + +def rotations(tile: Tile) -> Iterator[Tile]: + yield tile + for __ in range(3): + prev = deepcopy(tile) + for i in range(len(tile)): + for j in range(len(tile[i])): + prev[i][j] = tile[len(tile) - j - 1][i] + tile = prev + yield tile + + +def flips(tile: Tile) -> Iterator[Tile]: + yield tile + yield tile[::-1] + yield [line[::-1] for line in tile] + yield [line[::-1] for line in tile[::-1]] + + +def transforms(tile: Tile) -> List[Tile]: + # Can't use a set with lists... + res: List[Tile] = functools.reduce( + lambda l, it: l + [it] if not any(it == other for other in l) else l, + itertools.chain.from_iterable(map(rotations, flips(tile))), + [], + ) + return res + + +def borders(tile: Tile) -> Iterator[str]: + # They should match on same axis (i.e: go clockwise and counter-clockwise) + return map( + lambda s: "".join(s), + [ + tile[0], + [line[-1] for line in tile], + tile[-1], + [line[0] for line in tile], + ], + ) + + +Transforms = Dict[int, List[Tile]] +Tiling = List[List[Tuple[int, int]]] # Which tile, and which transform + + +def build_tiling(tiles: Tiles) -> Tuple[Transforms, Tiling]: + transform_mapping = {num: transforms(tile) for num, tile in tiles.items()} + trans_borders: Dict[int, Dict[int, List[str]]] = defaultdict(dict) + for num, trans in transform_mapping.items(): + for i, t in enumerate(trans): + trans_borders[num][i] = list(borders(t)) + length = math.isqrt(len(tiles)) + tiling = [[(-1, -1)] * length for __ in range(length)] + + def rec(x: int = 0, y: int = 0, used: Set[int] = set()) -> bool: + if x == length: + return True + for num, borders in trans_borders.items(): + if num in used: + continue + used |= {num} + + for trans, border in borders.items(): + top, __, __, left = border + if y > 0: + nid, ntrans = tiling[x][y - 1] + __, right, __, __ = trans_borders[nid][ntrans] + if left != right: + continue + if x > 0: + nid, ntrans = tiling[x - 1][y] + __, __, bottom, __ = trans_borders[nid][ntrans] + if top != bottom: + continue + tiling[x][y] = num, trans + + next_x, next_y = (x, y + 1) if y < (length - 1) else (x + 1, 0) + if rec(next_x, next_y, used): + return True + + used -= {num} + + tiling[x][y] = (-1, -1) + return False + + assert rec() + return transform_mapping, tiling + + +def solve(tiles: Tiles) -> int: + transforms, tiling = build_tiling(tiles) + return tiling[0][0][0] * tiling[0][-1][0] * tiling[-1][0][0] * tiling[-1][-1][0] + + +def main() -> None: + input = parse(sys.stdin.read()) + print(solve(input)) + + +if __name__ == "__main__": + main()