aboutsummaryrefslogtreecommitdiff
path: root/aoc2023/src/day1/solve.gleam
diff options
context:
space:
mode:
Diffstat (limited to 'aoc2023/src/day1/solve.gleam')
-rw-r--r--aoc2023/src/day1/solve.gleam78
1 files changed, 78 insertions, 0 deletions
diff --git a/aoc2023/src/day1/solve.gleam b/aoc2023/src/day1/solve.gleam
new file mode 100644
index 0000000..a1092b1
--- /dev/null
+++ b/aoc2023/src/day1/solve.gleam
@@ -0,0 +1,78 @@
+import adglent.{First, Second}
+import gleam/io
+import gleam/list
+import gleam/string
+import gleam/regex
+import gleam/result
+import gleam/int
+
+fn parse_digits(input: String) {
+ let assert Ok(re) = regex.from_string("[0-9]")
+
+ input
+ |> string.split("\n")
+ |> list.map(fn(s) {
+ let matches =
+ regex.scan(s, with: re)
+ |> list.map(fn(m) { m.content })
+
+ case matches {
+ [one] -> int.parse(one <> one)
+ _ ->
+ int.parse(
+ result.unwrap(list.first(matches), "") <> result.unwrap(
+ list.last(matches),
+ "",
+ ),
+ )
+ }
+ })
+}
+
+pub fn part1(input: String) {
+ input
+ |> parse_digits
+ |> result.values
+ |> int.sum
+ |> string.inspect
+}
+
+const substitutions = [
+ #("one", "o1e"),
+ #("two", "t2o"),
+ #("three", "t3e"),
+ #("four", "4"),
+ #("five", "5e"),
+ #("six", "6"),
+ #("seven", "7n"),
+ #("eight", "e8t"),
+ #("nine", "n9e"),
+ #("zero", "0o"),
+]
+
+pub fn part2(input: String) {
+ list.fold(
+ over: substitutions,
+ from: input,
+ with: fn(acc, sub) {
+ let #(from, to) = sub
+ string.replace(in: acc, each: from, with: to)
+ },
+ )
+ |> part1
+}
+
+pub fn main() {
+ let assert Ok(part) = adglent.get_part()
+ let assert Ok(input) = adglent.get_input("1")
+ case part {
+ First ->
+ part1(input)
+ |> adglent.inspect
+ |> io.println
+ Second ->
+ part2(input)
+ |> adglent.inspect
+ |> io.println
+ }
+}