mix format
This commit is contained in:
parent
14378b50b6
commit
839613668c
1 changed files with 59 additions and 34 deletions
|
@ -1,10 +1,12 @@
|
|||
defmodule Mix.Tasks.Day7 do
|
||||
use Mix.Task
|
||||
|
||||
def run(_) do
|
||||
IO.stream() |> solve() |> IO.puts()
|
||||
end
|
||||
@card_ordering_1 [?A, ?K, ?Q, ?J, ?T, ?9, ?8, ?7, ?6, ?5, ?4, ?3, ?2]
|
||||
@card_ordering_2 [?A, ?K, ?Q, ?T, ?9, ?8, ?7, ?6, ?5, ?4, ?3, ?2, ?J]
|
||||
|
||||
@card_ordering_1 [?A, ?K, ?Q, ?J, ?T, ?9, ?8, ?7, ?6, ?5, ?4, ?3, ?2]
|
||||
@card_ordering_2 [?A, ?K, ?Q, ?T, ?9, ?8, ?7, ?6, ?5, ?4, ?3, ?2, ?J]
|
||||
@cards ["A", "K", "Q", "T", "9", "8", "7", "6", "5", "4", "3", "2"]
|
||||
|
||||
def sort_by_hand_ascending(h1, h2, part) do
|
||||
|
@ -28,7 +30,7 @@ defmodule Mix.Tasks.Day7 do
|
|||
end
|
||||
|
||||
defp compare_cards(c1, c2, part) do
|
||||
ordering = if part === :part1, do: @card_ordering_1, else: @card_ordering_2
|
||||
ordering = if part === :part1, do: @card_ordering_1, else: @card_ordering_2
|
||||
c1_idx = get_card_index(c1, ordering)
|
||||
c2_idx = get_card_index(c2, ordering)
|
||||
|
||||
|
@ -45,57 +47,79 @@ defmodule Mix.Tasks.Day7 do
|
|||
other -> other
|
||||
end
|
||||
end
|
||||
|
||||
|
||||
defp generate_all_hands(h1) do
|
||||
cond do
|
||||
h1 |> String.contains?("J") -> for card <- @cards,
|
||||
h1 |> String.contains?("J") ->
|
||||
for card <- @cards,
|
||||
new_card <- generate_all_hands(h1 |> String.replace("J", card, global: false)) do
|
||||
new_card
|
||||
end
|
||||
|
||||
new_card <- generate_all_hands(h1 |> String.replace("J", card, global: false)) do
|
||||
new_card
|
||||
end
|
||||
true -> [h1]
|
||||
true ->
|
||||
[h1]
|
||||
end
|
||||
end
|
||||
|
||||
defp brute_force_best_hand(h1) do
|
||||
best = generate_all_hands(h1)
|
||||
best =
|
||||
generate_all_hands(h1)
|
||||
|> Enum.sort(fn cand1, cand2 ->
|
||||
|
||||
hand_type(cand1, :part2) > hand_type(cand2, :part2)
|
||||
end)
|
||||
|> Enum.at(0)
|
||||
best |> hand_type(:part2)
|
||||
|
||||
best |> hand_type(:part2)
|
||||
end
|
||||
|
||||
defp hand_type(h1, part) do
|
||||
cache = Process.get(:cache_key, %{})
|
||||
|
||||
store_and_return = fn i ->
|
||||
if part === :part2 do
|
||||
Process.put(:cache_key, cache |> Map.put(h1, i))
|
||||
end
|
||||
|
||||
i
|
||||
end
|
||||
if part === :part2 and cache |> Map.has_key?(h1) do
|
||||
|
||||
if part === :part2 and cache |> Map.has_key?(h1) do
|
||||
Map.get(cache, h1)
|
||||
else
|
||||
values = h1 |> String.graphemes() |> Enum.frequencies() |> Map.values()
|
||||
|
||||
cond do
|
||||
part === :part2 and h1 |> String.contains?("J") -> store_and_return.(brute_force_best_hand(h1))
|
||||
part === :part2 and h1 |> String.contains?("J") ->
|
||||
store_and_return.(brute_force_best_hand(h1))
|
||||
|
||||
# five of kind
|
||||
values |> Enum.any?(&(&1 === 5)) -> store_and_return.(7)
|
||||
values |> Enum.any?(&(&1 === 5)) ->
|
||||
store_and_return.(7)
|
||||
|
||||
# four of kind
|
||||
values |> Enum.any?(&(&1 === 4)) -> store_and_return.(6)
|
||||
values |> Enum.any?(&(&1 === 4)) ->
|
||||
store_and_return.(6)
|
||||
|
||||
# full house
|
||||
values |> Enum.any?(&(&1 === 3)) &&
|
||||
values |> Enum.any?(&(&1 === 2)) -> store_and_return.(5)
|
||||
values |> Enum.any?(&(&1 === 2)) ->
|
||||
store_and_return.(5)
|
||||
|
||||
# three
|
||||
values |> Enum.any?(&(&1 === 3)) -> store_and_return.(4)
|
||||
values |> Enum.any?(&(&1 === 3)) ->
|
||||
store_and_return.(4)
|
||||
|
||||
# two pair
|
||||
values |> Enum.filter(&(&1 === 2)) |> Enum.count() === 2 -> store_and_return.(3)
|
||||
values |> Enum.filter(&(&1 === 2)) |> Enum.count() === 2 ->
|
||||
store_and_return.(3)
|
||||
|
||||
# one
|
||||
values |> Enum.filter(&(&1 === 2)) |> Enum.count() === 1 -> store_and_return.(2)
|
||||
values |> Enum.filter(&(&1 === 2)) |> Enum.count() === 1 ->
|
||||
store_and_return.(2)
|
||||
|
||||
# high card
|
||||
values |> Enum.all?(&(&1 === 1)) -> store_and_return.(1)
|
||||
values |> Enum.all?(&(&1 === 1)) ->
|
||||
store_and_return.(1)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
@ -103,19 +127,20 @@ defmodule Mix.Tasks.Day7 do
|
|||
def solve(input) do
|
||||
part = :part2
|
||||
Process.delete(:key_cache)
|
||||
|
||||
input
|
||||
|> Stream.reject(&(&1 === ""))
|
||||
|> Stream.map(fn line ->
|
||||
line |> String.trim() |> String.split(" ") |> List.to_tuple()
|
||||
end)
|
||||
|> Enum.sort(fn {h1, _}, {h2, _} -> sort_by_hand_ascending(h1, h2, part) end)
|
||||
|> Stream.with_index()
|
||||
|> Enum.reduce(
|
||||
0,
|
||||
fn {{_, bid}, i}, acc ->
|
||||
bid = String.to_integer(bid)
|
||||
acc + bid * (i+1)
|
||||
end
|
||||
)
|
||||
|> Stream.reject(&(&1 === ""))
|
||||
|> Stream.map(fn line ->
|
||||
line |> String.trim() |> String.split(" ") |> List.to_tuple()
|
||||
end)
|
||||
|> Enum.sort(fn {h1, _}, {h2, _} -> sort_by_hand_ascending(h1, h2, part) end)
|
||||
|> Stream.with_index()
|
||||
|> Enum.reduce(
|
||||
0,
|
||||
fn {{_, bid}, i}, acc ->
|
||||
bid = String.to_integer(bid)
|
||||
acc + bid * (i + 1)
|
||||
end
|
||||
)
|
||||
end
|
||||
end
|
||||
|
|
Loading…
Add table
Reference in a new issue