あらざらむこの世のほかの思ひ出に今ひとたぴのあふこともがな

Advent Calendar 2022 108日目¹の記事です。
I'm looking forward to 12/25,2022
私のAdvent Calendar 2022 一覧。

はじめに

この記事は、Advent Of Code 2021 Day 7: The Treachery of WhalesをElixirで楽しんでみます。

私はGitHubでログインしました。

私の回答

私の回答です。
解けるには解けましたが、効率が悪いし、メモリをたくさん使っているようにおもいます。

私の回答

input = "3,4,3,1,2"

Part 1

list = input |> String.split(",", trim: true) |> Enum.map(&String.to_integer/1)

count = Enum.count(list)
max = Enum.max(list)

0..max |> Enum.reduce(count * max, fn pos, min_fuel ->
  {largers, smallers} = Enum.split_with(list, & &1 >= pos)
  fuel = Enum.sum(largers) - pos * Enum.count(largers) + pos * Enum.count(smallers) - Enum.sum(smallers)
  if fuel < min_fuel, do: fuel, else: min_fuel
end)

Part 2

0..max |> Enum.reduce(Enum.sum(1..max) * count, fn pos, min_fuel ->
  fuel = Enum.reduce(list, 0, fn i, acc -> Enum.sum(0..abs(i - pos)) + acc end)
  if fuel < min_fuel, do: fuel, else: min_fuel
end)