Day 4: Ceres Search
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)
- You can send code in code blocks by using three backticks, the code, and then three backticks or use something such as https://topaz.github.io/paste/ if you prefer sending it through a URL
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
Uiua
This one was nice. The second part seemed quite daunting at first but wasn’t actually that hard in the end.
Run with example input here
Row ← ⌕ "XMAS" RevRow ← ⌕"SAMX" Sum ← /+/+ Count ← +∩Sum⊃Row RevRow PartOne ← ( &rs ∞ &fo "input-4.txt" ⊜∘≠@\n. ⊙+⟜∩Count⟜⍉ # horizontal and vertical search ⟜(/+⧈(Count⍉≡⬚@ ↻⇡⧻.)4) /+⧈(Count⍉≡⬚@ ↻¯⇡⧻.)4 ++ ) Mask ← °⊚×2⇡5 # Create variations of X-MAS Vars ← ( ["M S" " A " "M S"] ≡♭[∩⟜⍉]≡⇌. Mask ⊏0⊞▽¤ ) PartTwo ← ( &rs ∞ &fo "input-4.txt" ⊜∘≠@\n. ⧈(/+♭⊞≍⊙¤Vars▽Mask♭)3_3 Sum ) &p "Day 4:" &pf "Part 1: " &p PartOne &pf "Part 2: " &p PartTwo
Lisp
Not super happy with the code, but it got the job done.
Part 1 and 2
(defun p1-process-line (line) (to-symbols line)) (defun found-word-h (word data i j) "checks for a word existing from the point horizontally to the right" (loop for j2 from j for w in word when (not (eql w (aref data i j2))) return nil finally (return t))) (defun found-word-v (word data i j) "checks for a word existing from the point vertically down" (loop for i2 from i for w in word when (not (eql w (aref data i2 j))) return nil finally (return t))) (defun found-word-d-l (word data i j) "checks for a word existsing from the point diagonally to the left and down" (destructuring-bind (n m) (array-dimensions data) (declare (ignorable n)) (and (>= (- i (length word)) -1) (>= m (+ j (length word))) (loop for i2 from i downto 0 for j2 from j for w in word when (not (eql w (aref data i2 j2))) return nil finally (return t))))) (defun found-word-d-r (word data i j) "checks for a word existing from the point diagonally to the right and down" (destructuring-bind (n m) (array-dimensions data) (and (>= n (+ i (length word))) (>= m (+ j (length word))) (loop for i2 from i for j2 from j for w in word when (not (eql w (aref data i2 j2))) return nil finally (return t))) )) (defun count-word-h (data word) "Counts horizontal matches of the word" (let ((word-r (reverse word)) (word-l (length word))) (destructuring-bind (n m) (array-dimensions data) (loop for i from 0 below n sum (loop for j from 0 upto (- m word-l) count (found-word-h word data i j) count (found-word-h word-r data i j)))))) (defun count-word-v (data word) "Counts vertical matches of the word" (let ((word-r (reverse word)) (word-l (length word))) (destructuring-bind (n m) (array-dimensions data) (loop for j from 0 below m sum (loop for i from 0 upto (- n word-l) count (found-word-v word data i j) count (found-word-v word-r data i j)))))) (defun count-word-d (data word) "Counts diagonal matches of the word" (let ((word-r (reverse word))) (destructuring-bind (n m) (array-dimensions data) (loop for i from 0 below n sum (loop for j from 0 below m count (found-word-d-l word data i j) count (found-word-d-l word-r data i j) count (found-word-d-r word data i j) count (found-word-d-r word-r data i j) ))))) (defun run-p1 (file) "cares about the word xmas in any direction" (let ((word '(X M A S)) (data (list-to-2d-array (read-file file #'p1-process-line)))) (+ (count-word-v data word) (count-word-h data word) (count-word-d data word)))) (defun run-p2 (file) "cares about an x of mas crossed with mas" (let ((word '(M A S)) (word-r '(S A M)) (data (list-to-2d-array (read-file file #'p1-process-line)))) (destructuring-bind (n m) (array-dimensions data) (loop for i from 0 below (- n 2) sum (loop for j from 0 below (- m 2) count (and (found-word-d-r word data i j) (found-word-d-l word data (+ i 2) j)) count (and (found-word-d-r word-r data i j) (found-word-d-l word data (+ i 2) j)) count (and (found-word-d-r word data i j) (found-word-d-l word-r data (+ i 2) j)) count (and (found-word-d-r word-r data i j) (found-word-d-l word-r data (+ i 2) j)) )))))
Zig
const std = @import("std"); const List = std.ArrayList; const tokenizeScalar = std.mem.tokenizeScalar; const parseInt = std.fmt.parseInt; const print = std.debug.print; const eql = std.mem.eql; var gpa = std.heap.GeneralPurposeAllocator(.{}){}; const alloc = gpa.allocator(); const Point = struct { x: isize, y: isize, fn add(self: *const Point, point: *const Point) Point { return Point{ .x = self.x + point.x, .y = self.y + point.y }; } }; // note: i have no idea how to use this or if it's even possible // const DirectionType = enum(u8) { Up, Down, Left, Right, UpLeft, UpRight, DownLeft, DownRight }; // const Direction = union(DirectionType) { // up: Point = .{ .x = 0, .y = 0 }, // }; const AllDirections = [_]Point{ .{ .x = 0, .y = -1 }, // up .{ .x = 0, .y = 1 }, // down .{ .x = -1, .y = 0 }, // left .{ .x = 1, .y = 0 }, // right .{ .x = -1, .y = -1 }, // up left .{ .x = 1, .y = -1 }, // up right .{ .x = -1, .y = 1 }, // down left .{ .x = 1, .y = 1 }, // down right }; const Answer = struct { xmas: u32, mas: u32, }; pub fn searchXmas(letters: List([]const u8), search_char: u8, position: Point, direction: Point) u32 { const current_char = getChar(letters, position); if (current_char == search_char) { const next = position.add(&direction); if (current_char == 'M') { return searchXmas(letters, 'A', next, direction); } else if (current_char == 'A') { return searchXmas(letters, 'S', next, direction); } else if (current_char == 'S') { return 1; // found all letters } } return 0; } pub fn countXmas(letters: List([]const u8), starts: List(Point)) u32 { var counter: u32 = 0; for (starts.items) |start| { for (AllDirections) |direction| { const next = start.add(&direction); counter += searchXmas(letters, 'M', next, direction); } } return counter; } pub fn countMas(letters: List([]const u8), starts: List(Point)) u32 { var counter: u32 = 0; for (starts.items) |start| { const a_char = getChar(letters, start) orelse continue; const top_left_char = getChar(letters, start.add(&AllDirections[4])) orelse continue; const down_right_char = getChar(letters, start.add(&AllDirections[7])) orelse continue; const top_right_char = getChar(letters, start.add(&AllDirections[5])) orelse continue; const down_left_char = getChar(letters, start.add(&AllDirections[6])) orelse continue; const tldr = [3]u8{ top_left_char, a_char, down_right_char }; const trdl = [3]u8{ top_right_char, a_char, down_left_char }; if ((eql(u8, &tldr, "MAS") or eql(u8, &tldr, "SAM")) and (eql(u8, &trdl, "MAS") or eql(u8, &trdl, "SAM"))) { counter += 1; } } return counter; } pub fn getChar(letters: List([]const u8), point: Point) ?u8 { if (0 > point.x or point.x >= letters.items.len) { return null; } const row = @as(usize, @intCast(point.x)); if (0 > point.y or point.y >= letters.items[row].len) { return null; } const col = @as(usize, @intCast(point.y)); return letters.items[row][col]; } pub fn solve(input: []const u8) !Answer { var rows = tokenizeScalar(u8, input, '\n'); var letters = List([]const u8).init(alloc); defer letters.deinit(); var x_starts = List(Point).init(alloc); defer x_starts.deinit(); var a_starts = List(Point).init(alloc); defer a_starts.deinit(); var x: usize = 0; while (rows.next()) |row| { try letters.append(row); for (row, 0..) |letter, y| { if (letter == 'X') { try x_starts.append(.{ .x = @intCast(x), .y = @intCast(y) }); } else if (letter == 'A') { try a_starts.append(.{ .x = @intCast(x), .y = @intCast(y) }); } } x += 1; } // PART 1 const xmas = countXmas(letters, x_starts); // PART 2 const mas = countMas(letters, a_starts); return Answer{ .xmas = xmas, .mas = mas }; } pub fn main() !void { const answer = try solve(@embedFile("input.txt")); print("Part 1: {d}\n", .{answer.xmas}); print("Part 2: {d}\n", .{answer.mas}); } test "test input" { const answer = try solve(@embedFile("test.txt")); try std.testing.expectEqual(18, answer.xmas); }
Elixir
defmodule AdventOfCode.Solution.Year2024.Day04 do use AdventOfCode.Solution.SharedParse defmodule Map do defstruct [:chars, :width, :height] end @impl true def parse(input) do chars = String.split(input, "\n", trim: true) |> Enum.map(&String.codepoints/1) %Map{chars: chars, width: length(Enum.at(chars, 0)), height: length(chars)} end def at(%Map{} = map, x, y) do cond do x < 0 or x >= map.width or y < 0 or y >= map.height -> "" true -> map.chars |> Enum.at(y, []) |> Enum.at(x, "") end end def part1(map) do dirs = for dx <- -1..1, dy <- -1..1, {dx, dy} != {0, 0}, do: {dx, dy} xmas = String.codepoints("XMAS") |> Enum.with_index() |> Enum.drop(1) for x <- 0..(map.width - 1), y <- 0..(map.height - 1), "X" == at(map, x, y), {dx, dy} <- dirs, xmas |> Enum.all?(fn {c, n} -> at(map, x + dx * n, y + dy * n) == c end), reduce: 0 do t -> t + 1 end end def part2(map) do for x <- 0..(map.width - 1), y <- 0..(map.height - 1), "A" == at(map, x, y), (at(map, x - 1, y - 1) <> at(map, x + 1, y + 1)) in ["MS", "SM"], (at(map, x - 1, y + 1) <> at(map, x + 1, y - 1)) in ["MS", "SM"], reduce: 0 do t -> t + 1 end end end
Haskell
Popular language this year :)
I got embarrassingly stuck on this one trying to be clever with list operations. Then I realized I should just use an array…
import Data.Array.Unboxed (UArray) import Data.Array.Unboxed qualified as A import Data.Bifunctor readInput :: String -> UArray (Int, Int) Char readInput s = let rows = lines s n = length rows in A.listArray ((1, 1), (n, n)) $ concat rows s1 `eq` s2 = s1 == s2 || s1 == reverse s2 part1 arr = length $ filter isXmas $ concatMap lines $ A.indices arr where isXmas ps = all (A.inRange $ A.bounds arr) ps && map (arr A.!) ps `eq` "XMAS" lines p = [take 4 $ iterate (bimap (+ di) (+ dj)) p | (di, dj) <- [(1, 0), (0, 1), (1, 1), (1, -1)]] part2 arr = length $ filter isXmas innerPoints where innerPoints = let ((i1, j1), (i2, j2)) = A.bounds arr in [(i, j) | i <- [i1 + 1 .. i2 - 1], j <- [j1 + 1 .. j2 - 1]] isXmas p = up p `eq` "MAS" && down p `eq` "MAS" up (i, j) = map (arr A.!) [(i + 1, j - 1), (i, j), (i - 1, j + 1)] down (i, j) = map (arr A.!) [(i - 1, j - 1), (i, j), (i + 1, j + 1)] main = do input <- readInput <$> readFile "input04" print $ part1 input print $ part2 input
I struggled a lot more when doing list slices that I would’ve liked to
Haskell
import Data.List qualified as List collectDiagonal :: [String] -> Int -> Int -> String collectDiagonal c y x | length c > y && length (c !! y) > x = c !! y !! x : collectDiagonal c (y+1) (x+1) | otherwise = [] part1 c = do let forwardXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails) $ c let backwardXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails . reverse) $ c let downwardXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails ) . List.transpose $ c let upwardXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails . reverse ) . List.transpose $ c let leftSideDiagonals = map (\ y -> collectDiagonal c y 0) [0..length c] let leftTopDiagonals = map (\ x -> collectDiagonal c 0 x) [1..(length . List.head $ c)] let leftDiagonals = leftSideDiagonals ++ leftTopDiagonals let rightSideDiagonals = map (\ y -> collectDiagonal (map List.reverse c) y 0) [0..length c] let rightTopDiagonals = map (\ x -> collectDiagonal (map List.reverse c) 0 x) [1..(length . List.head $ c)] let rightDiagonals = rightSideDiagonals ++ rightTopDiagonals let diagonals = leftDiagonals ++ rightDiagonals let diagonalXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails) $ diagonals let reverseDiagonalXMAS = map (length . filter (List.isPrefixOf "XMAS") . List.tails . reverse) $ diagonals print . sum $ [sum forwardXMAS, sum backwardXMAS, sum downwardXMAS, sum upwardXMAS, sum diagonalXMAS, sum reverseDiagonalXMAS] return () getBlock h w c y x = map (take w . drop x) . take h . drop y $ c isXBlock b = do let diagonal1 = collectDiagonal b 0 0 let diagonal2 = collectDiagonal (map List.reverse b) 0 0 diagonal1 `elem` ["SAM", "MAS"] && diagonal2 `elem` ["SAM", "MAS"] part2 c = do let lineBlocks = List.map (getBlock 3 3 c) [0..length c - 1] let groupedBlocks = List.map (flip List.map [0..(length . head $ c) - 1]) lineBlocks print . sum . map (length . filter isXBlock) $ groupedBlocks return () main = do c <- lines <$> getContents part1 c part2 c return ()
Nim
Could be done more elegantly, but I haven’t bothered yet.
proc solve(input: string): AOCSolution[int, int] = var lines = input.splitLines() block p1: # horiz for line in lines: for i in 0..line.high-3: if line[i..i+3] in ["XMAS", "SAMX"]: inc result.part1 for y in 0..lines.high-3: #vert for x in 0..lines[0].high: let word = collect(for y in y..y+3: lines[y][x]) if word in [@"XMAS", @"SAMX"]: inc result.part1 #diag \ for x in 0..lines[0].high-3: let word = collect(for d in 0..3: lines[y+d][x+d]) if word in [@"XMAS", @"SAMX"]: inc result.part1 #diag / for x in 3..lines[0].high: let word = collect(for d in 0..3: lines[y+d][x-d]) if word in [@"XMAS", @"SAMX"]: inc result.part1 block p2: for y in 0..lines.high-2: for x in 0..lines[0].high-2: let diagNW = collect(for d in 0..2: lines[y+d][x+d]) let diagNE = collect(for d in 0..2: lines[y+d][x+2-d]) if diagNW in [@"MAS", @"SAM"] and diagNE in [@"MAS", @"SAM"]: inc result.part2
J
Unsurprisingly this is the kind of problem that J is really good at. The dyadic case (table) of the adverb
/
is doing all the heavy lifting here: it makes a higher rank tensor by traversing items of the specified rank on each side and combining them according to the remaining frame of each side’s shape. The hard part is arranging the arguments so that your resulting matrix has its axes in the correct order.data_file_name =: '4.data' NB. cutopen yields boxed lines, so unbox them and ravel items to make a letter matrix grid =: ,. > cutopen fread data_file_name NB. pad the grid on every side with #'XMAS' - 1 spaces hpadded_grid =: ((' ' & ,) @: (, & ' '))"1 grid padded_grid =: (3 1 $ ' ') , hpadded_grid , (3 1 $ ' ') NB. traversal vectors directions =: 8 2 $ 1 0 1 1 0 1 _1 1 _1 0 _1 _1 0 _1 1 _1 NB. rpos cpos matches rdir cdir if the string starting at rpos cpos in NB. direction rdir cdir is the string we want matches =: 4 : 0 */ ,'XMAS' -: padded_grid {~ <"1 x +"1 y *"1 0 i. 4 )"1 positions =: (3 + i. 0 { $ grid) ,"0/ (3 + i. 1 { $ grid) result1 =: +/, positions matches/ directions NB. pairs of traversal vectors x_directions =: 4 2 2 $ 1 1 _1 1 1 1 1 _1 _1 _1 _1 1 _1 _1 1 _1 NB. rpos cpos x_matches 2 2 $ rdir1 cdir1 rdir2 cdir2 if there is an 'A' at NB. rpos cpos and the string in each of dir1 and dir2 centered at rpos cpos NB. is the string we want x_matches =: 4 : 0 NB. (2 2 $ rdir1 cdir1 rdir2 cdir2) *"1 0/ (_1 + i.3) yields a matrix NB. 2 3 $ (_1 * dir1) , (0 * dir1) , (1 * dir1) followed by the same for dir2 */ ,'MAS' -:"1 padded_grid {~ <"1 x +"1 y *"1 0/ _1 + i. 3 )"1 2 result2 =: +/, positions x_matches/ x_directions
Uiua
Just part1 for now as I need to walk the dog :-)
[edit] Part 2 now added, and a nicer approach than Part 1 in my opinion, if you’re able to keep that many dimensions straight in your head :-)
[edit 2] Tightened it up a bit more.
Grid ← ⊜∘⊸≠@\n "MMMSXXMASM\nMSAMXMSMSA\nAMXSXMAAMM\nMSAMASMSMX\nXMASAMXAMM\nXXAMMXXAMA\nSMSMSASXSS\nSAXAMASAAA\nMAMMMXMMMM\nMXMXAXMASX" ≡⍉⍉×⇡4¤[1_0 0_1 1_1 1_¯1] # Use core dirs to build sets of 4-offsets. ↯∞_2⇡△ Grid # Get all possible starting points. &p/+♭⊞(+∩(≍"XMAS")⇌.⬚@.⊡:Grid≡+¤) # Part 1. Join the two into a table, use to pick 4-elements, check, count. Diags ← [[¯. 1_1] [¯. 1_¯1]] BothMas ← /×≡(+∩(≍"MS")⇌.)⬚@.⊡≡+Diags¤¤ # True if both diags here are MAS. &p/+≡BothMas⊚="A"⟜¤Grid # Part 2. For all "A"s in grid, check diags, count where good.
I’m not even sure how to write most of these characters
The operators have all got ascii names you can type, and the formatter converts them to the symbols. It’s a bit odd but really worthwhile, as you get access to the powerful array handling functionality that made solving today’s challenges so much more straightforward than in other languages.
It looks quite functional indeed
Factor
spoiler
: get-input ( -- rows ) "vocab:aoc-2024/04/input.txt" utf8 file-lines ; : verticals ( rows -- lines ) [ dimension last [0..b) ] keep cols ; : slash-origins ( rows -- coords ) dimension [ first [0..b) [ 0 2array ] map ] [ first2 [ 1 - ] [ 1 (a..b] ] bi* [ 2array ] with map ] bi append ; : backslash-origins ( rows -- coords ) dimension first2 [ [0..b) [ 0 2array ] map ] [ 1 (a..b] [ 0 swap 2array ] map ] bi* append ; : slash ( rows origin -- line ) first2 [ 0 [a..b] ] [ pick dimension last [a..b) ] bi* zip swap matrix-nths ; : backslash ( rows origin -- line ) [ dup dimension ] dip first2 [ over first [a..b) ] [ pick last [a..b) ] bi* zip nip swap matrix-nths ; : slashes ( rows -- lines ) dup slash-origins [ slash ] with map ; : backslashes ( rows -- lines ) dup backslash-origins [ backslash ] with map ; : word-count ( line word -- n ) dupd [ reverse ] dip '[ _ subseq-indices length ] bi@ + ; : part1 ( -- n ) get-input { [ ] [ verticals ] [ slashes ] [ backslashes ] } cleave-array concat [ "XMAS" word-count ] map-sum ; : origin-adistances ( rows origins line-quot: ( rows origin -- line ) -- origin-adistances-assoc ) with zip-with "MAS" "SAM" [ '[ [ _ subseq-indices ] map-values ] ] bi@ bi append harvest-values [ [ 1 + ] map ] map-values ; inline : a-coords ( origin-adistances coord-quot: ( adistance -- row-delta col-delta ) -- coords ) '[ first2 [ @ 2array v+ ] with map ] map-concat ; inline : slash-a-coords ( rows -- coords ) dup slash-origins [ slash ] origin-adistances [ [ 0 swap - ] keep ] a-coords ; : backslash-a-coords ( rows -- coords ) dup backslash-origins [ backslash ] origin-adistances [ dup ] a-coords ; : part2 ( -- n ) get-input [ slash-a-coords ] [ backslash-a-coords ] bi intersect length ;
Better viewed on GitHub.
Haskell
import Control.Arrow import Data.Array.Unboxed import Data.List type Pos = (Int, Int) type Board = Array Pos Char data Dir = N | NE | E | SE | S | SW | W | NW target = "XMAS" parse s = listArray ((1, 1), (n, m)) [l !! i !! j | i <- [0 .. n - 1], j <- [0 .. m - 1]] where l = lines s (n, m) = (length $ head l, length l) move N = first pred move S = first succ move E = second pred move W = second succ move NW = move N . move W move SW = move S . move W move NE = move N . move E move SE = move S . move E check :: Board -> Pos -> Int -> Dir -> Bool check b p i d = i >= length target || ( inRange (bounds b) p && (b ! p) == (target !! i) && check b (move d p) (succ i) d ) checkAllDirs :: Board -> Pos -> Int checkAllDirs b p = length . filter (check b p 0) $ [N, NE, E, SE, S, SW, W, NW] check2 :: Board -> Pos -> Bool check2 b p = all (inRange (bounds b)) moves && ((b ! p) == 'A') && ("SSMM" `elem` rotations) where rotations = rots $ (b !) <$> moves moves = flip move p <$> [NE, SE, SW, NW] rots xs = init $ zipWith (++) (tails xs) (inits xs) part1 b = sum $ checkAllDirs b <$> indices b part2 b = length . filter (check2 b) $ indices b main = getContents >>= print . (part1 &&& part2) . parse
C
What can I say, bunch of for loops! I add a 3 cell border to avoid having to do bounds checking in the inner loops.
Code
#include "common.h" #define GZ 146 int main(int argc, char **argv) { static char g[GZ][GZ]; static const char w[] = "XMAS"; int p1=0,p2=0, x,y, m,i; if (argc > 1) DISCARD(freopen(argv[1], "r", stdin)); for (y=3; y<GZ && fgets(g[y]+3, GZ-3, stdin); y++) ; for (y=3; y<GZ-3; y++) for (x=3; x<GZ-3; x++) { for (m=1,i=0; i<4; i++) {m &= g[y+i][x]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y-i][x]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y][x+i]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y][x-i]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y+i][x+i]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y-i][x-i]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y+i][x-i]==w[i];} p1+=m; for (m=1,i=0; i<4; i++) {m &= g[y-i][x+i]==w[i];} p1+=m; p2 += g[y+1][x+1]=='A' && ((g[y][x] =='M' && g[y+2][x+2]=='S') || (g[y][x] =='S' && g[y+2][x+2]=='M')) && ((g[y+2][x]=='M' && g[y][x+2] =='S') || (g[y+2][x]=='S' && g[y][x+2] =='M')); } printf("04: %d %d\n", p1, p2); }
C#
public class Day04 : Solver { private int width, height; private char[,] data; public void Presolve(string input) { var lines = input.Trim().Split("\n").ToList(); height = lines.Count; width = lines[0].Length; data = new char[height, width]; for (int i = 0; i < height; i++) { for (int j = 0; j < width; j++) { data[i, j] = lines[i][j]; } } } private static readonly string word = "XMAS"; public string SolveFirst() { int counter = 0; for (int start_i = 0; start_i < height; start_i++) { for (int start_j = 0; start_j < width; start_j++) { if (data[start_i, start_j] != word[0]) continue; for (int di = -1; di <= 1; di++) { for (int dj = -1; dj <= 1; dj++) { if (di == 0 && dj == 0) continue; int end_i = start_i + di * (word.Length - 1); int end_j = start_j + dj * (word.Length - 1); if (end_i < 0 || end_j < 0 || end_i >= height || end_j >= width) continue; for (int k = 1; k < word.Length; k++) { if (data[start_i + di * k, start_j + dj * k] != word[k]) break; if (k == word.Length - 1) counter++; } } } } } return counter.ToString(); } public string SolveSecond() { int counter = 0; for (int start_i = 1; start_i < height - 1; start_i++) { for (int start_j = 1; start_j < width - 1; start_j++) { if (data[start_i, start_j] != 'A') continue; int even_mas_starts = 0; for (int di = -1; di <= 1; di++) { for (int dj = -1; dj <= 1; dj++) { if (di == 0 && dj == 0) continue; if ((di + dj) % 2 != 0) continue; if (data[start_i + di, start_j + dj] != 'M') continue; if (data[start_i - di, start_j - dj] != 'S') continue; even_mas_starts++; } } if (even_mas_starts == 2) counter++; } } return counter.ToString(); } }
Nim
import ../aoc, strutils type Cell* = tuple[x,y:int] #the 8 grid direction const directions : array[8, Cell] = [ (1, 0), (-1, 0), (0, 1), ( 0,-1), (1, 1), (-1,-1), (1,-1), (-1, 1) ] const xmas = "XMAS" #part 1 proc searchXMAS*(grid:seq[string], x,y:int):int = #search in all 8 directions (provided we can find a full match in that direction) let w = grid[0].len let h = grid.len for dir in directions: # check if XMAS can even fit let xEnd = x + dir.x * 3 let yEnd = y + dir.y * 3 if xEnd < 0 or xEnd >= w or yEnd < 0 or yEnd >= h: continue; #step along direction var matches = 0 for s in 0..3: if grid[y + dir.y * s][x + dir.x * s] == xmas[s]: inc matches if matches == xmas.len: inc result #part 2 proc isMAS(grid:seq[string], c, o:Cell):bool= let ca : Cell = (c.x+o.x, c.y+o.y) let cb : Cell = (c.x-o.x, c.y-o.y) let a = grid[ca.y][ca.x] let b = grid[cb.y][cb.x] (a == 'M' and b == 'S') or (a == 'S' and b == 'M') proc searchCrossMAS*(grid:seq[string], x,y:int):bool = grid[y][x] == 'A' and grid.isMAS((x,y), (1,1)) and grid.isMAS((x,y), (1,-1)) proc solve*(input:string): array[2,int] = let grid = input.splitLines let w = grid[0].len let h = grid.len #part 1 for y in 0..<h: for x in 0..<w: result[0] += grid.searchXMAS(x, y) #part 2, skipping borders for y in 1..<h-1: for x in 1..<w-1: result[1] += (int)grid.searchCrossMAS(x, y)
Part 1 was done really quickly. Part 2 as well, but the result was not accepted…
Turns out +MAS isn’t actually a thing :P
Rust
Ugh. Spent way too long on today’s. Should have just used my own grid structure from last year. I will likely refactor to use that. Even though it’s likely a super slow implementation, the convenience of dealing with it is better than shoehorning in the
grid::Grid<T>
from that crate.solution (no supporting code)
use grid::Grid; use crate::shared::{ grid2d::{iter_diag_nesw, iter_diag_nwse, Point}, util::read_lines, }; fn parse_grid(input: &[String]) -> Grid<u8> { let cols = input.first().unwrap().len(); Grid::from_vec( input .iter() .flat_map(|row| row.chars().map(|c| c as u8).collect::<Vec<u8>>()) .collect(), cols, ) } fn part1(grid: &Grid<u8>) -> usize { let mut xmas_count = 0; let rows = grid .iter_rows() .map(|d| String::from_utf8(d.copied().collect()).unwrap()); let cols = grid .iter_cols() .map(|d| String::from_utf8(d.copied().collect()).unwrap()); for diag in iter_diag_nesw(grid) .chain(iter_diag_nwse(grid)) .filter_map(|d| { if d.len() >= 4 { Some(String::from_utf8(d.clone()).unwrap()) } else { None } }) .chain(rows) .chain(cols) { xmas_count += diag.matches("XMAS").count() + diag.matches("SAMX").count() } xmas_count } fn part2(grid: &Grid<u8>) -> usize { let mut xmas_count = 0; let valid = [ [b'M', b'M', b'S', b'S'], [b'M', b'S', b'S', b'M'], [b'S', b'M', b'M', b'S'], [b'S', b'S', b'M', b'M'], ]; for x in 1..grid.cols() - 1 { for y in 1..grid.rows() - 1 { if grid.get(y, x) == Some(&b'A') && valid.contains( &(Point::new(x as isize, y as isize) .diagonal_neighbors(grid) .map(|i| i.unwrap_or(0))), ) { xmas_count += 1; } } } xmas_count } pub fn solve() { let input = read_lines("inputs/day04.txt"); let grid = parse_grid(&input); println!("Part 1: {}", part1(&grid)); println!("Part 2: {}", part2(&grid)); }
And here’s a link to the Github if you care to see the gross supporting code :D