Day 21

Mix.install([:kino_aoc])

Section

{:ok, puzzle_input} =
  KinoAOC.download_puzzle("2024", "21", System.fetch_env!("LB_ADVENT_OF_CODE_SESSION"))

{:ok, "638A\n965A\n780A\n803A\n246A"}

codes =
  puzzle_input
  |> String.split("\n", trim: true)
  |> Enum.map(fn <<num::3-binary>> <> "A" = code ->
    {String.to_charlist(code), String.to_integer(num)}
  end)

[{~c"638A", 638}, {~c"965A", 965}, {~c"780A", 780}, {~c"803A", 803}, {~c"246A", 246}]

dirpad = %{
  [?A, ?>] => ~c[vA],
  [?A, ?^] => ~c[<A],
  [?A, ?v] => ~c[<vA],
  [?A, ?<] => ~c[v<<A],

  [?>, ?A] => ~c[^A],
  [?>, ?^] => ~c[<^A],
  [?>, ?v] => ~c[<A],
  [?>, ?<] => ~c[<<A],

  [?^, ?A] => ~c[>A],
  [?^, ?>] => ~c[v>A],
  [?^, ?v] => ~c[vA],
  [?^, ?<] => ~c[v<A],

  [?<, ?A] => ~c[>>^A],
  [?<, ?^] => ~c[>^A],
  [?<, ?v] => ~c[>A],
  [?<, ?>] => ~c[>>A],

  [?v, ?A] => ~c[>^A],
  [?v, ?^] => ~c[^A],
  [?v, ?<] => ~c[<A],
  [?v, ?>] => ~c[>A]
}

%{
  ~c"<>" => ~c">>A",
  ~c"<A" => ~c">>^A",
  ~c"<^" => ~c">^A",
  ~c"<v" => ~c">A",
  ~c"><" => ~c"<<A",
  ~c">A" => ~c"^A",
  ~c">^" => ~c"<^A",
  ~c">v" => ~c"<A",
  ~c"A<" => ~c"v<<A",
  ~c"A>" => ~c"vA",
  ~c"A^" => ~c"<A",
  ~c"Av" => ~c"<vA",
  ~c"^<" => ~c"v<A",
  ~c"^>" => ~c"v>A",
  ~c"^A" => ~c">A",
  ~c"^v" => ~c"vA",
  ~c"v<" => ~c"<A",
  ~c"v>" => ~c">A",
  ~c"vA" => ~c">^A",
  ~c"v^" => ~c"^A"
}

nil

defmodule Keypad do
  def naive(input, lut) do
    [?A | input]
    |> Enum.chunk_every(2, 1, :discard)
    |> Enum.flat_map(&Map.get(lut, &1, [?A]))
  end

  def expand(input, lut) do
    Stream.concat([?A], input)
    |> Stream.chunk_every(2, 1, :discard)
    |> Stream.flat_map(&Map.get(lut, &1, [?A]))
  end
end

{:module, Keypad, <<70, 79, 82, 49, 0, 0, 9, ...>>, {:expand, 2}}

nil

Keypad.expand(~c[<A^A>^^AvvvA], dirpad)
|> Enum.count()

#|> Enum.map(fn {k, v} ->
#  [?A | Map.get(dirpad, k, [?A])]
#  |> Enum.chunk_every(2, 1, :discard)
#  |> Enum.map()
#end)
#|> Enum.group_by(&elem(&1, 0), &elem(&1, 1))
#|> Map.new(fn {k, v} -> {k, Enum.sum(v)} end)

28

Keypad.naive(~c[<A^A>^^AvvvA], dirpad)
|> Keypad.naive(dirpad)
|> Enum.chunk_every(2, 1, [?A])
|> Enum.frequencies()
|> Map.values()
|> Enum.sum()

68

~c"<vA<AA>>^AvAA<^A>Av<<A>>^AvA^A<vA>^Av<<A>^A>AAvA^Av<<A>A>^AAAvA<^A>A"
~c"<vA<AA>>^AvAA<^A>A<v<A>>^AvA^A<vA>^A<v<A>^A>AAvA^A<v<A>A>^AAAvA<^A>A" |> length

68