From 1c1bc2ea6e5b45b7cb7377cd5e15645318bf1647 Mon Sep 17 00:00:00 2001
From: Alexandre Jesus <adbjesus@gmail.com>
Date: Sat, 7 Dec 2024 16:10:00 +0000
Subject: Day 7 (and small update to day 6)

---
 src/day07.exs | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 77 insertions(+)
 create mode 100644 src/day07.exs

(limited to 'src/day07.exs')

diff --git a/src/day07.exs b/src/day07.exs
new file mode 100644
index 0000000..1adeb54
--- /dev/null
+++ b/src/day07.exs
@@ -0,0 +1,77 @@
+defmodule Utils do
+  def parse_integer_list(string, split, trim \\ true) do
+    string
+    |> String.split(split, trim: trim)
+    |> Enum.map(& String.to_integer(&1))
+  end
+end
+
+defmodule Equation do
+  @derive Inspect
+  defstruct [:result, :terms]
+
+  def from_string(string) do
+    string
+    |> Utils.parse_integer_list([":", " "])
+    |> then(& %Equation{result: hd(&1), terms: tl(&1)})
+  end
+
+  def has_lr_solution(equation, ops) do
+    has_lr_solution(hd(equation.terms), tl(equation.terms), equation.result, ops)
+  end
+
+  defp has_lr_solution(acc, terms, result, ops) do
+    cond do
+      acc > result ->
+        false
+      terms == [] and acc == result ->
+        true
+      terms == [] ->
+        false
+      true ->
+        ops
+        |> Enum.any?(& has_lr_solution(apply_op(acc, hd(terms), &1), tl(terms), result, ops))
+    end
+  end
+
+  defp apply_op(l, r, op) do
+    case op do
+      :add ->
+        l + r
+      :mul ->
+        l * r
+      :concat ->
+        String.to_integer("#{l}#{r}")
+    end
+  end
+end
+
+defmodule Day07 do
+  def part1(data) do
+    data
+    |> String.split("\n", trim: true)
+    |> Enum.map(& Equation.from_string(&1))
+    |> Enum.filter(& Equation.has_lr_solution(&1, [:mul, :add]))
+    |> Enum.map(& &1.result)
+    |> Enum.sum()
+  end
+
+  def part2(data) do
+    data
+    |> String.split("\n", trim: true)
+    |> Enum.map(& Equation.from_string(&1))
+    |> Enum.filter(& Equation.has_lr_solution(&1, [:mul, :concat, :add]))
+    |> Enum.map(& &1.result)
+    |> Enum.sum()
+  end
+end
+
+data = IO.read(:stdio, :eof)
+
+{time1 , ans1} = :timer.tc(fn -> Day07.part1(data) end)
+IO.puts("Time  : #{time1 / 1000000}")
+IO.puts("Answer: #{ans1}")
+
+{time2 , ans2} = :timer.tc(fn -> Day07.part2(data) end)
+IO.puts("Time  : #{time2 / 1000000}")
+IO.puts("Answer: #{ans2}")
-- 
cgit v1.2.3