diff options
author | H.J <thechairman@thechairman.info> | 2024-10-09 11:36:55 -0400 |
---|---|---|
committer | H.J <thechairman@thechairman.info> | 2024-10-09 11:36:55 -0400 |
commit | 8777ff071f7bb37631baa7b6717ad29961e50911 (patch) | |
tree | 6d59c4ed58e454b960339c3d1151f0a879e8d7cb /gleam/aoc2019/src/aoc_2019/day_2.gleam | |
parent | 6156a9ef7be4012063a042aafb4e9b0d7eadde8e (diff) | |
download | gleam_aoc-8777ff071f7bb37631baa7b6717ad29961e50911.tar.gz gleam_aoc-8777ff071f7bb37631baa7b6717ad29961e50911.zip |
sorting by language
Diffstat (limited to 'gleam/aoc2019/src/aoc_2019/day_2.gleam')
-rw-r--r-- | gleam/aoc2019/src/aoc_2019/day_2.gleam | 81 |
1 files changed, 81 insertions, 0 deletions
diff --git a/gleam/aoc2019/src/aoc_2019/day_2.gleam b/gleam/aoc2019/src/aoc_2019/day_2.gleam new file mode 100644 index 0000000..8faa0ea --- /dev/null +++ b/gleam/aoc2019/src/aoc_2019/day_2.gleam @@ -0,0 +1,81 @@ +import gary.{type ErlangArray} +import gary/array.{type ArrayError} +import gleam/bool +import gleam/int +import gleam/list +import gleam/result +import gleam/string + +pub fn parse(input: String) -> ErlangArray(Int) { + input + |> string.split(",") + |> list.map(int.parse) + |> result.values() + |> array.from_list(default: -1) + |> array.make_fixed() +} + +pub fn pt_1(input: ErlangArray(Int)) -> Int { + let assert Ok(result) = + input + |> edit_starting_intcodes(12, 2) + |> run_intcode(0) + + result +} + +pub fn pt_2(input: ErlangArray(Int)) -> Int { + let assert [result] = { + use noun <- list.flat_map(list.range(0, 99)) + use verb <- list.filter_map(list.range(0, 99)) + let result = input |> edit_starting_intcodes(noun, verb) |> run_intcode(0) + case result == Ok(19_690_720) { + True -> Ok(100 * noun + verb) + False -> Error(Nil) + } + } + + result +} + +fn run_intcode( + intcode: ErlangArray(Int), + pointer: Int, +) -> Result(Int, ArrayError) { + let assert Ok(op_code) = array.get(intcode, pointer) + let op = get_op(op_code) + + use <- bool.guard(result.is_error(op), array.get(intcode, 0)) + let assert Ok(position_1) = array.get(intcode, pointer + 1) + let assert Ok(position_2) = array.get(intcode, pointer + 2) + let assert Ok(position_3) = array.get(intcode, pointer + 3) + + let assert Ok(value_1) = array.get(intcode, position_1) + let assert Ok(value_2) = array.get(intcode, position_2) + + let assert Ok(f) = op + let new_value = f(value_1, value_2) + let assert Ok(updated_intcode) = array.set(intcode, position_3, new_value) + run_intcode(updated_intcode, pointer + 4) +} + +fn edit_starting_intcodes( + intcodes: ErlangArray(Int), + new_code_1: Int, + new_code_2: Int, +) -> ErlangArray(Int) { + let assert Ok(updated) = + intcodes + |> array.set(at: 1, put: new_code_1) + |> result.try(array.set(into: _, at: 2, put: new_code_2)) + updated +} + +fn get_op(code: Int) -> Result(fn(Int, Int) -> Int, Nil) { + case code { + 1 -> Ok(int.add) + 2 -> Ok(int.multiply) + 99 -> Error(Nil) + _ -> panic as "bad opcode" + } +} |