77 lines
2.5 KiB
Nim
77 lines
2.5 KiB
Nim
import ./common, std/[sequtils, algorithm, sugar, sets, strformat, strutils, tables, options, json, hashes]
|
|
|
|
type
|
|
Point = (int, int)
|
|
Image = object
|
|
pixels: TableRef[Point, bool]
|
|
topLeft: Point
|
|
bottomRight: Point
|
|
infinite: bool
|
|
|
|
proc parse(s: openArray[string]): Image =
|
|
var pixels = newTable[Point, bool]()
|
|
for y,l in s.pairs:
|
|
for x,c in l:
|
|
pixels[(x, y)] = c == '#'
|
|
return Image(topLeft: (0, 0), bottomRight: (s[0].len-1, s.len-1), pixels: pixels, infinite: false)
|
|
|
|
proc enhance(image: var Image, algo: string) =
|
|
var (tlx, tly) = image.topLeft
|
|
var (brx, bry) = image.bottomRight
|
|
var newPixels = newTable[Point, bool]()
|
|
var xr = tlx-1..brx+1
|
|
var yr = tly-1..bry+1
|
|
for x in xr:
|
|
for y in yr:
|
|
var sample = 0
|
|
for sy in y-1..y+1:
|
|
for sx in x-1..x+1:
|
|
if image.pixels.getOrDefault((sx, sy), image.infinite):
|
|
inc sample
|
|
sample = sample shl 1
|
|
sample = sample shr 1
|
|
# echo &"{x}, {y} => {sample}"
|
|
newPixels[(x, y)] = algo[sample] == '#'
|
|
if x < tlx: tlx = x
|
|
if y < tly: tly = y
|
|
if x > brx: brx = x
|
|
if y > bry: bry = y
|
|
image.topLeft = (tlx, tly)
|
|
image.bottomRight = (brx, bry)
|
|
image.pixels = newPixels
|
|
if image.infinite:
|
|
image.infinite = algo[^1] == '#'
|
|
else:
|
|
image.infinite = algo[0] == '#'
|
|
|
|
proc `$`(image: Image): string =
|
|
var s: seq[string] = @[]
|
|
var (tlx, tly) = image.topLeft
|
|
var (brx, bry) = image.bottomRight
|
|
for _ in tly..bry:
|
|
s.add ".".repeat((brx-tlx)+1)
|
|
for p in image.pixels.keys:
|
|
let (x, y) = p
|
|
if image.pixels[p]: s[y-tly][x-tlx] = '#'
|
|
s.join("\n")
|
|
|
|
proc solver(input: Lines, times: int): uint64 =
|
|
let algo = input[0]
|
|
var image = input[2..^1].parse()
|
|
for _ in 1..times: image.enhance algo
|
|
for v in image.pixels.values:
|
|
if v: inc result
|
|
|
|
proc p1(input: Lines): uint64 = input.solver(2)
|
|
proc p2(input: Lines): uint64 = input.solver(50)
|
|
|
|
const input = """
|
|
..#.#..#####.#.#.#.###.##.....###.##.#..###.####..#####..#....#..#..##..###..######.###...####..#..#####..##..#.#####...##.#.#..#.##..#.#......#.###.######.###.####...#.##.##..#..#..#####.....#.#....###..#.##......#.....#..#..#..##..#...##.######.####.####.#.#...#.......#..#.#.#...####.##.#......#..#...##.#.##..#...##.#.##..###.#......#.#.......#.#.#.####.###.##...#.....####.#..#..#.##.#....##..#.####....##...##..#...#......#.#.......#.......##..####..#...#.#.#...##..#.#..###..#####........#..####......#..#
|
|
|
|
#..#.
|
|
#....
|
|
##..#
|
|
..#..
|
|
..### """.strip().split('\n')
|
|
doDayX 20, (n: int) => n.loadInput, p1, p2, (input, 35'u64, 3351'u64)
|