This commit is contained in:
Felix Van der Jeugt 2021-12-20 13:13:11 +01:00
parent f4227e42ec
commit fadc49c09d
No known key found for this signature in database
GPG Key ID: 58B209295023754D
3 changed files with 215 additions and 0 deletions

64
day20/part1.lua Executable file
View File

@ -0,0 +1,64 @@
#!/usr/bin/env luajit
require("utils")
require("grid")
local f = io.open(arg[3])
local line = f:read("*l")
local enhancer = {}
for char in chars(line) do
if char == "#" then
table.insert(enhancer, 1)
else
table.insert(enhancer, 0)
end
end
f:read("*l")
line = f:read("*l")
local image = grid(1, 1, #line, #line, 0)
local r, c = 0, 0
while line ~= nil do
r = r + 1
c = 0
for char in chars(line) do
c = c + 1
if char == "#" then
image:set(r, c, 1)
end
end
line = f:read("*l")
end
function zoom(image, enhancer)
local newdefault = enhancer[image.default * (#enhancer - 1) + 1]
local newimage = grid(image.top - 1, image.left - 1,
image.height + 2, image.width + 2, newdefault)
for v, r, c in image:iter() do
newimage:set(r, c, v)
end
return newimage
end
function enhance(image, enhancer)
local newimage = zoom(image, enhancer)
for _, r, c in newimage:iter() do
local index = 0
for v, _, _ in image:neigh9(r, c) do
index = index * 2 + v
end
newimage:set(r, c, enhancer[index + 1])
end
return newimage
end
local function pprint(c)
if c == 0
then return "."
else return "#"
end
end
local counter = 0
for v in enhance(enhance(image, enhancer), enhancer):iter() do
counter = counter + v
end
print(counter)

68
day20/part2.lua Executable file
View File

@ -0,0 +1,68 @@
#!/usr/bin/env luajit
require("utils")
require("grid")
local f = io.open(arg[3])
local line = f:read("*l")
local enhancer = {}
for char in chars(line) do
if char == "#" then
table.insert(enhancer, 1)
else
table.insert(enhancer, 0)
end
end
f:read("*l")
line = f:read("*l")
local image = grid(1, 1, #line, #line, 0)
local r, c = 0, 0
while line ~= nil do
r = r + 1
c = 0
for char in chars(line) do
c = c + 1
if char == "#" then
image:set(r, c, 1)
end
end
line = f:read("*l")
end
function zoom(image, enhancer)
local newdefault = enhancer[image.default * (#enhancer - 1) + 1]
local newimage = grid(image.top - 1, image.left - 1,
image.height + 2, image.width + 2, newdefault)
for v, r, c in image:iter() do
newimage:set(r, c, v)
end
return newimage
end
function enhance(image, enhancer)
local newimage = zoom(image, enhancer)
for _, r, c in newimage:iter() do
local index = 0
for v, _, _ in image:neigh9(r, c) do
index = index * 2 + v
end
newimage:set(r, c, enhancer[index + 1])
end
return newimage
end
local function pprint(c)
if c == 0
then return "."
else return "#"
end
end
for i = 1, 50 do
image = enhance(image, enhancer)
end
local counter = 0
for v in image:iter() do
counter = counter + v
end
print(counter)

83
grid.lua Normal file
View File

@ -0,0 +1,83 @@
local function set(self, r, c, v)
if r < self.top or r > self.bottom or c < self.left or c > self.right then
error(string.format("index (%d %d) out of bounds (%d..%d %d..%d)",
r, c, self.top, self.bottom, self.left, self.right))
end
self.grid[r * self.width + c] = v
end
local function get(self, r, c)
if r < self.top or r > self.bottom or c < self.left or c > self.right then
if self.default then
return self.default
else
error(string.format("index (%d %d) out of bounds (%d..%d %d..%d)",
r, c, self.top, self.bottom, self.left, self.right))
end
end
local item = self.grid[r * self.width + c]
if item == nil and self.default then
return self.default
else
return item
end
end
local function iter(self, r, c, h, w)
if r == nil then r = self.top end
if c == nil then c = self.left end
if h == nil then h = self.height end
if w == nil then w = self.width end
local r0, c0 = r, c
return function()
if r0 > r + h - 1 then return nil end
local rb, cb = r0, c0
c0 = c0 + 1
if c0 > c + w - 1 then
c0 = c
r0 = r0 + 1
end
return self:get(rb, cb), rb, cb
end
end
local function neigh9(self, r, c, exclude)
local iter = self:iter(r - 1, c - 1, 3, 3)
return function()
local i, ri, ci = iter()
if exclude and ri == r and ci == c then
local i, ri, ci = iter()
end
return i, ri, ci
end
end
local function write(self, format)
if not format then format = function(i)
return string.format("%3s", i)
end end
for r = self.top, self.bottom do
for c = self.left, self.right do
io.write(format(self:get(r, c)))
end
io.write("\n")
end
end
function grid(top, left, width, height, default)
return {
top = top,
left = left,
height = height,
width = width,
bottom = top + height - 1,
right = left + width - 1,
default = default,
grid = {},
get = get,
set = set,
iter = iter,
neigh9 = neigh9,
write = write,
}
end