闘魂Elixir ── Advent of code 2023 Day 9 Part 2 を Livebook で楽しむ

$\huge{元氣ですかーーーーッ！！！}$
$\huge{元氣があればなんでもできる！}$

$\huge{闘魂とは己に打ち克つこと。}$
$\huge{そして闘いを通じて己の魂を磨いていく}$
$\huge{ことだと思います}$

はじめに

@torifukukaiou さんの パク リスペクト記事です

Elixir Livebook で Advent of Code 2023 の問題を解いてみます

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

問題はこちら

Part 1 はこちら

セットアップ

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

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

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

入力の取得

Day 9 の入力を取得します

私の答え

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

details

回答用のモジュールです

入力を扱いやすい形にするための parse と、回答を作るための resolve 関数を持っています

defmodule Resolver do
  def parse(input) do
    input
    |> String.split("\n")
    |> Enum.map(fn line ->
      line
      |> String.split(" ")
      |> Enum.map(&String.to_integer(&1))
    end)
  end

  def resolve(array_list) do
    array_list
    |> Enum.map(fn array ->
      diff_array_list =
        0..(Enum.count(array) - 1)
        |> Enum.reduce_while([array], fn _index, acc ->
          last_array = Enum.at(acc, -1)
          diff_array = diff(last_array)

          if Enum.all?(diff_array, &(&1 == 0)) do
            {:halt, acc ++ [diff_array]}
          else
            {:cont, acc ++ [diff_array]}
          end
        end)

      even_sum =
        diff_array_list
        |> Enum.take_every(2)
        |> Enum.map(&hd(&1))
        |> Enum.sum()

      odd_sum =
        diff_array_list
        |> tl()
        |> Enum.take_every(2)
        |> Enum.map(&hd(&1))
        |> Enum.sum()

      even_sum - odd_sum
    end)
    |> Enum.sum()
  end

  defp diff(array) do
    {_, a_array} = List.pop_at(array, 0)
    {_, b_array} = List.pop_at(array, -1)

    Enum.zip(a_array, b_array)
    |> Enum.map(fn {a, b} ->
      a - b
    end)
  end
end

parse は Part 1 と同じです

resolve も差数列を求めるところまでは同じです

Part 1 は次の数字を予測しましたが、 Part 2 では前の数値を予測します

前の数値は以下のようにして求めました

• even_sum: 偶数番目の差数列の先頭の合計
• odd_sum: 奇数番目の差数列の先頭の合計
• 前の数値: even_sum - odd_sum

計算方法さえわかれば簡単ですね

まとめ

Part 2 の実装はほとんど Part 1 と変わらないので簡単でした

どちらかというと数学的なセンスを試されている感じですね