ruidy/ai_advent_code_2024
1
0
Fork 0
mirror of https://github.com/rjNemo/ai_advent_code_2024 synced 2026-06-10 04:26:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: Add type specs and simplify calculate_similarity_score function

Browse source
This commit is contained in:
Ruidy 2024-12-01 17:59:45 +01:00 committed by Ruidy (aider)
parent 5a5a16ab1c
commit 5a966b4839
No known key found for this signature in database
GPG key ID: E00F51288CB857CC
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
lib/advent_code2024/day1.ex
View file

@ -3,6 +3,7 @@ defmodule AdventCode2024.Day1 do
Solution for Advent of Code 2024 - Day 1: Historian Hysteria
"""
@spec solve(String.t()) :: {:ok, integer()} | {:error, String.t()}
def solve(input_file \\ "day1/input.txt") do
case File.read(input_file) do
{:ok, content} ->
@ -15,6 +16,7 @@ defmodule AdventCode2024.Day1 do
end
end
@spec solve_part2(String.t()) :: {:ok, integer()} | {:error, String.t()}
def solve_part2(input_file \\ "day1/input.txt") do
case File.read(input_file) do
{:ok, content} ->
@ -27,6 +29,7 @@ defmodule AdventCode2024.Day1 do
end
end
@spec parse_input(String.t()) :: {[integer()], [integer()]}
defp parse_input(content) do
{left, right} =
content
@ -40,19 +43,19 @@ defmodule AdventCode2024.Day1 do
{left, right}
end
@spec calculate_total_distance([integer()], [integer()]) :: integer()
defp calculate_total_distance(left, right) do
Enum.zip(Enum.sort(left), Enum.sort(right))
|> Enum.map(fn {a, b} -> abs(a - b) end)
|> Enum.sum()
end
@spec calculate_similarity_score([integer()], [integer()]) :: integer()
defp calculate_similarity_score(left, right) do
frequencies = Enum.frequencies(right)
left
|> Enum.map(fn num ->
num * Map.get(frequencies, num, 0)
end)
|> Enum.map(&(&1 * Map.get(frequencies, &1, 0)))
|> Enum.sum()
end
end