Advent of code 2024 Day 4 Part 1 を Livebook で楽しむ

はじめに

Advent of code 2024 Day 4 の Part 1 を解きます

Part 1 と Part 2 で大きく解法が異なるため分割しました

問題文はこちら

実装したノートブックはこちら

セットアップ

Kino AOC をインストールします

Mix.install([
  {:kino_aoc, "~> 0.1"}
])

Kino AOC の使い方はこちらを参照

入力の取得

"Advent of Code Helper" スマートセルを追加し、 Day 4 の入力を取得します

私の答え

私の答えです。
折りたたんでおきます。
▶を押して開いてください。

回答

まずは小さい入力例で解いてみます

input =
  """
  MMMSXXMASM
  MSAMXMSMSA
  AMXSXMAAMM
  MSAMASMSMX
  XMASAMXAMM
  XXAMMXXAMA
  SMSMSASXSS
  SAXAMASAAA
  MAMMMXMMMM
  MXMXAXMASX
  """

右から左、左から右に読むパターンは単純に正規表現で取得できます

Regex.scan(~r/XMAS/, input)

実行結果

[["XMAS"], ["XMAS"], ["XMAS"]]

Regex.scan(~r/SAMX/, input)

実行結果

[["SAMX"], ["SAMX"]]

上下方向は縦横を入れ替えれば取得可能です

turned_input =
  input
  |> String.trim()
  |> String.split("\n")
  |> Enum.map(&String.codepoints(&1))
  |> Enum.zip()
  |> Enum.map(fn col -> col |> Tuple.to_list() |> Enum.join() end)
  |> Enum.join("\n")

IO.puts(turned_input)

標準出力

MMAMXXSSMM
MSMSMXMAAX
MAXAAASXMM
SMSMSMMAMX
XXXAAMSMMA
XMMSMXAAXX
MSAMXXSSMM
AMASAAXAMA
SSMMMMSAMS
MAMXMASAMX

Regex.scan(~r/XMAS/, turned_input)

Regex.scan(~r/SAMX/, turned_input)

斜め方向を取得するために、まずは横にずらします

turned_input =
  input
  |> String.trim()
  |> String.split("\n")
  |> Enum.with_index()
  |> Enum.map(fn {row, index} ->
    String.duplicate(".", index) <> row <> String.duplicate(".", String.length(row) - index - 1)
  end)
  |> Enum.join("\n")

IO.puts(turned_input)

標準出力

MMMSXXMASM.........
.MSAMXMSMSA........
..AMXSXMAAMM.......
...MSAMASMSMX......
....XMASAMXAMM.....
.....XXAMMXXAMA....
......SMSMSASXSS...
.......SAXAMASAAA..
........MAMMMXMMMM.
.........MXMXAXMASX

これを回転させることで、斜め方向に読むことができます

turned_input =
  input
  |> String.trim()
  |> String.split("\n")
  |> Enum.with_index()
  |> Enum.map(fn {row, index} ->
    String.duplicate(".", index) <> row <> String.duplicate(".", String.length(row) - index - 1)
  end)
  |> Enum.map(&String.codepoints(&1))
  |> Enum.zip()
  |> Enum.map(fn col -> col |> Tuple.to_list() |> Enum.join() end)
  |> Enum.join("\n")

IO.puts(turned_input)

標準出力

M.........
MM........
MSA.......
SAMM......
XMXSX.....
XXSAMX....
MMXMAXS...
ASMASAMS..
SMASAMSAM.
MSAMMMMXAM
.AMSXXSAMX
..MMAXAMMM
...XMASAMX
....MMXSXA
.....ASAMX
......SAMM
.......AMA
........MS
.........X

Regex.scan(~r/XMAS/, turned_input)

Regex.scan(~r/SAMX/, turned_input)

ずらす向きを逆にすれば逆の斜め方向も読むことができます

ここまでのことを踏まえ、ある方向に対して XMAS の存在する数を取得する関数を用意します

search_xmas = fn input ->
  IO.puts("")
  IO.puts(input)
  Enum.sum([
    Regex.scan(~r/XMAS/, input) |> length(),
    Regex.scan(~r/SAMX/, input) |> length()
  ])
end

search_xmas.(input)

実行結果

5

全ての方向に対して XMAS を探す関数を用意します

search_all_xmas = fn input ->
  Enum.sum([
    input
    |> String.trim()
    |> search_xmas.(),
    input
    |> String.trim()
    |> String.split("\n")
    |> Enum.map(&String.codepoints(&1))
    |> Enum.zip()
    |> Enum.map(fn col -> col |> Tuple.to_list() |> Enum.join() end)
    |> Enum.join("\n")
    |> search_xmas.(),
    input
    |> String.trim()
    |> String.split("\n")
    |> Enum.with_index()
    |> Enum.map(fn {row, index} ->
      String.duplicate(".", index) <> row <> String.duplicate(".", String.length(row) - index - 1)
    end)
    |> Enum.map(&String.codepoints(&1))
    |> Enum.zip()
    |> Enum.map(fn col -> col |> Tuple.to_list() |> Enum.join() end)
    |> Enum.join("\n")
    |> search_xmas.(),
    input
    |> String.trim()
    |> String.split("\n")
    |> Enum.with_index()
    |> Enum.map(fn {row, index} ->
      String.duplicate(".", String.length(row) - index - 1) <> row <> String.duplicate(".", index)
    end)
    |> Enum.map(&String.codepoints(&1))
    |> Enum.zip()
    |> Enum.map(fn col -> col |> Tuple.to_list() |> Enum.join() end)
    |> Enum.join("\n")
    |> search_xmas.()
  ])
end

search_all_xmas.(input)

標準出力

MMMSXXMASM
MSAMXMSMSA
AMXSXMAAMM
MSAMASMSMX
XMASAMXAMM
XXAMMXXAMA
SMSMSASXSS
SAXAMASAAA
MAMMMXMMMM
MXMXAXMASX

MMAMXXSSMM
MSMSMXMAAX
MAXAAASXMM
SMSMSMMAMX
XXXAAMSMMA
XMMSMXAAXX
MSAMXXSSMM
AMASAAXAMA
SSMMMMSAMS
MAMXMASAMX

M.........
MM........
MSA.......
SAMM......
XMXSX.....
MMXMAXS...
ASMASAMS..
SMASAMSAM.
MSAMMMMXAM
.AMSXXSAMX
..MMAXAMMM
...XMASAMX
....MMXSXA
.....ASAMX
......SAMM
.......AMA
........MS
.........X

.........M
........MX
.......SAM
......SAMX
.....XMXMA
....XXSAMX
...MMAMMXM
..ASAMSAMA
.MMASMASMS
MSXMAXSAMX
MASAMXXAM.
MMXSXASA..
SXMMAMS...
XMASMA....
XSAMM.....
MMMX......
ASM.......
SA........
M.........

実行結果

18

実際の入力に対して実行すれば答えが出ます

search_all_xmas.(puzzle_input)

まとめ

問題文から ChatGPT に画像を生成してもらいました

文字列を斜めに読むのが面白い問題でした

Part 2 はこちら