56 lines
1.6 KiB
Python
Executable file
56 lines
1.6 KiB
Python
Executable file
#!/usr/bin/env python
|
|
|
|
import sys
|
|
from typing import NamedTuple
|
|
|
|
|
|
class Point(NamedTuple):
|
|
x: int
|
|
y: int
|
|
|
|
|
|
class Region(NamedTuple):
|
|
size: Point
|
|
pieces: list[int]
|
|
|
|
@classmethod
|
|
def from_str(cls, input: str) -> "Region":
|
|
raw_size, *raw_pieces = input.split(" ")
|
|
size = Point(*map(int, raw_size.removesuffix(":").split("x")))
|
|
pieces = [int(n) for n in raw_pieces]
|
|
return cls(size, pieces)
|
|
|
|
|
|
def solve(input: str) -> int:
|
|
def parse_shape(input: list[str]) -> set[Point]:
|
|
return {
|
|
Point(x, y)
|
|
for x, line in enumerate(input)
|
|
for y, c in enumerate(line)
|
|
if c == "#"
|
|
}
|
|
|
|
def parse(input: str) -> tuple[list[set[Point]], list[Region]]:
|
|
*raw_shapes, raw_regions = input.split("\n\n")
|
|
shapes = [parse_shape(shape.splitlines()[1:]) for shape in raw_shapes]
|
|
regions = [Region.from_str(region) for region in raw_regions.splitlines()]
|
|
return shapes, regions
|
|
|
|
def can_fit_stupid(shapes: list[set[Point]], region: Region) -> bool:
|
|
# Does not actually try to fit the pieces, just checks the area
|
|
# This does *not* work on the sample input
|
|
minimum_size = sum(len(shapes[i]) * n for i, n in enumerate(region.pieces))
|
|
region_size = region.size.x * region.size.y
|
|
return minimum_size <= region_size
|
|
|
|
shapes, regions = parse(input)
|
|
return sum(can_fit_stupid(shapes, region) for region in regions)
|
|
|
|
|
|
def main() -> None:
|
|
input = sys.stdin.read()
|
|
print(solve(input))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|