From 3960278955eabae08f0636d7153aa9e2ae4fd318 Mon Sep 17 00:00:00 2001 From: Adam Millerchip Date: Tue, 29 Nov 2022 01:05:36 +0900 Subject: [PATCH] 2021 day6 part 1 --- 2021/day6.exs | 63 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 63 insertions(+) create mode 100644 2021/day6.exs diff --git a/2021/day6.exs b/2021/day6.exs new file mode 100644 index 0000000..bcdad7a --- /dev/null +++ b/2021/day6.exs @@ -0,0 +1,63 @@ +defmodule Day6 do + def part1(input) do + input + |> run_days(80) + |> length() + end + + def part2(_input) do + :ok + end + + def run_days(school, 0), do: school + def run_days(school, days), do: run_days(day(school), days - 1) + + def day(school), do: Enum.flat_map(school, &fish/1) + + def fish(0), do: [6, 8] + def fish(age), do: [age - 1] + + def input do + with [input_filename] <- System.argv(), + {:ok, input} <- File.read(input_filename) do + input + |> String.trim() + |> String.split(",") + |> Enum.map(&String.to_integer/1) + else + _ -> :error + end + end + + ####################### + # HERE BE BOILERPLATE # + ####################### + + def run do + case input() do + :error -> print_usage() + input -> run_parts_with_timer(input) + end + end + + defp run_parts_with_timer(input) do + run_with_timer(1, fn -> part1(input) end) + run_with_timer(2, fn -> part2(input) end) + end + + defp run_with_timer(part, fun) do + {time, result} = :timer.tc(fun) + IO.puts("Part #{part} (completed in #{format_time(time)}):\n") + IO.puts("#{result}\n") + end + + defp format_time(μsec) when μsec < 1_000, do: "#{μsec}μs" + defp format_time(μsec) when μsec < 1_000_000, do: "#{μsec / 1000}ms" + defp format_time(μsec), do: "#{μsec / 1_000_000}s" + + defp print_usage do + IO.puts("Usage: elixir day6.exs input_filename") + end +end + +# Day6.run()