Day 8: Seven Segment Search

Dependencies

Mix.install(
  [
    {:aoc_common, github: "rob-brown/AdventOfCode2021"},
    {:kino, "~> 0.4.0"}
  ],
  force: false
)

input = Kino.Input.textarea("Please paste your input file:")

Part 1

You barely reach the safety of the cave when the whale smashes into the cave mouth, collapsing it. Sensors indicate another exit to this cave at a much greater depth, so you have no choice but to press on.

As your submarine slowly makes its way through the cave system, you notice that the four-digit seven-segment displays in your submarine are malfunctioning; they must have been damaged during the escape. You’ll be in a lot of trouble without them, so you’d better figure out what’s wrong.

Each digit of a seven-segment display is rendered by turning on or off any of seven segments named a through g:

  0:      1:      2:      3:      4:
 aaaa    ....    aaaa    aaaa    ....
b    c  .    c  .    c  .    c  b    c
b    c  .    c  .    c  .    c  b    c
 ....    ....    dddd    dddd    dddd
e    f  .    f  e    .  .    f  .    f
e    f  .    f  e    .  .    f  .    f
 gggg    ....    gggg    gggg    ....

  5:      6:      7:      8:      9:
 aaaa    aaaa    aaaa    aaaa    aaaa
b    .  b    .  .    c  b    c  b    c
b    .  b    .  .    c  b    c  b    c
 dddd    dddd    ....    dddd    dddd
.    f  e    f  .    f  e    f  .    f
.    f  e    f  .    f  e    f  .    f
 gggg    gggg    ....    gggg    gggg

So, to render a 1, only segments c and f would be turned on; the rest would be off. To render a 7, only segments a, c, and f would be turned on.

The problem is that the signals which control the segments have been mixed up on each display. The submarine is still trying to display numbers by producing output on signal wires a through g, but those wires are connected to segments randomly. Worse, the wire/segment connections are mixed up separately for each four-digit display! (All of the digits within a display use the same connections, though.)

So, you might know that only signal wires b and g are turned on, but that doesn’t mean segments b and g are turned on: the only digit that uses two segments is 1, so it must mean segments c and f are meant to be on. With just that information, you still can’t tell which wire (b/g) goes to which segment (c/f). For that, you’ll need to collect more information.

For each display, you watch the changing signals for a while, make a note of all ten unique signal patterns you see, and then write down a single four digit output value (your puzzle input). Using the signal patterns, you should be able to work out which pattern corresponds to which digit.

For example, here is what you might see in a single entry in your notes:

acedgfb cdfbe gcdfa fbcad dab cefabd cdfgeb eafb cagedb ab |
cdfeb fcadb cdfeb cdbaf

(The entry is wrapped here to two lines so it fits; in your notes, it will all be on a single line.)

Each entry consists of ten unique signal patterns, a | delimiter, and finally the four digit output value. Within an entry, the same wire/segment connections are used (but you don’t know what the connections actually are). The unique signal patterns correspond to the ten different ways the submarine tries to render a digit using the current wire/segment connections. Because 7 is the only digit that uses three segments, dab in the above example means that to render a 7, signal lines d, a, and b are on. Because 4 is the only digit that uses four segments, eafb means that to render a 4, signal lines e, a, f, and b are on.

Using this information, you should be able to work out which combination of signal wires corresponds to each of the ten digits. Then, you can decode the four digit output value. Unfortunately, in the above example, all of the digits in the output value (cdfeb fcadb cdfeb cdbaf) use five segments and are more difficult to deduce.

For now, focus on the easy digits. Consider this larger example:

be cfbegad cbdgef fgaecd cgeb fdcge agebfd fecdb fabcd edb |
fdgacbe cefdb cefbgd gcbe
edbfga begcd cbg gc gcadebf fbgde acbgfd abcde gfcbed gfec |
fcgedb cgb dgebacf gc
fgaebd cg bdaec gdafb agbcfd gdcbef bgcad gfac gcb cdgabef |
cg cg fdcagb cbg
fbegcd cbd adcefb dageb afcb bc aefdc ecdab fgdeca fcdbega |
efabcd cedba gadfec cb
aecbfdg fbg gf bafeg dbefa fcge gcbea fcaegb dgceab fcbdga |
gecf egdcabf bgf bfgea
fgeab ca afcebg bdacfeg cfaedg gcfdb baec bfadeg bafgc acf |
gebdcfa ecba ca fadegcb
dbcfg fgd bdegcaf fgec aegbdf ecdfab fbedc dacgb gdcebf gf |
cefg dcbef fcge gbcadfe
bdfegc cbegaf gecbf dfcage bdacg ed bedf ced adcbefg gebcd |
ed bcgafe cdgba cbgef
egadfb cdbfeg cegd fecab cgb gbdefca cg fgcdab egfdb bfceg |
gbdfcae bgc cg cgb
gcafb gcf dcaebfg ecagb gf abcdeg gaef cafbge fdbac fegbdc |
fgae cfgab fg bagce

Because the digits 1, 4, 7, and 8 each use a unique number of segments, you should be able to tell which combinations of signals correspond to those digits. Counting only digits in the output values (the part after | on each line), in the above example, there are 26 instances of digits that use a unique number of segments (highlighted above).

In the output values, how many times do digits 1, 4, 7, or 8 appear?

defmodule Part1 do
  use AocCommon

  def run(input) do
    input
    |> String.trim_trailing()
    |> String.split("\n", trim: true)
    |> Enum.map(&count_1_4_7_8/1)
    |> Enum.sum()
  end

  defp count_1_4_7_8(line) do
    [_, output] = String.split(line, "|", parts: 2)

    output
    |> String.split(" ", trim: true)
    |> Enum.count(&(String.length(&1) in [2, 4, 3, 7]))
  end
end

input
|> Kino.Input.read()
|> Part1.run()

Part 2

Through a little deduction, you should now be able to determine the remaining digits. Consider again the first example above:

acedgfb cdfbe gcdfa fbcad dab cefabd cdfgeb eafb cagedb ab |
cdfeb fcadb cdfeb cdbaf

After some careful analysis, the mapping between signal wires and segments only make sense in the following configuration:

 dddd
e    a
e    a
 ffff
g    b
g    b
 cccc

So, the unique signal patterns would correspond to the following digits:

• acedgfb: 8
• cdfbe: 5
• gcdfa: 2
• fbcad: 3
• dab: 7
• cefabd: 9
• cdfgeb: 6
• eafb: 4
• cagedb: 0
• ab: 1

Then, the four digits of the output value can be decoded:

• cdfeb: 5
• fcadb: 3
• cdfeb: 5
• cdbaf: 3

Therefore, the output value for this entry is 5353.

Following this same process for each entry in the second, larger example above, the output value of each entry can be determined:

• fdgacbe cefdb cefbgd gcbe: 8394
• fcgedb cgb dgebacf gc: 9781
• cg cg fdcagb cbg: 1197
• efabcd cedba gadfec cb: 9361
• gecf egdcabf bgf bfgea: 4873
• gebdcfa ecba ca fadegcb: 8418
• cefg dcbef fcge gbcadfe: 4548
• ed bcgafe cdgba cbgef: 1625
• gbdfcae bgc cg cgb: 8717
• fgae cfgab fg bagce: 4315

Adding all of the output values in this larger example produces 61229.

For each entry, determine all of the wire/segment connections and decode the four-digit output values. What do you get if you add up all of the output values?

defmodule Part2 do
  use AocCommon

  def run(input) do
    all_wirings = permutations(~w[a b c d e f g])

    input
    |> String.trim_trailing()
    |> String.split("\n", trim: true)
    |> Enum.map(&solve_wiring(&1, all_wirings))
    |> Enum.sum()
  end

  defp solve_wiring(line, all_wirings) do
    [samples, outputs] = String.split(line, "|", parts: 2)

    samples =
      samples
      |> String.split(" ", trim: true)
      |> Enum.map(&String.graphemes/1)

    outputs =
      outputs
      |> String.split(" ", trim: true)
      |> Enum.map(&String.graphemes/1)

    samples
    |> find_wiring(all_wirings)
    |> calculate_output(outputs)
  end

  defp find_wiring(samples, all_wirings) do
    all_wirings
    |> Stream.filter(&valid_wiring?(samples, &1))
    |> Enum.at(0)
  end

  defp valid_wiring?(samples, wiring) do
    Enum.all?(samples, &(&1 |> to_segments(wiring) |> number() != :error))
  end

  defp calculate_output(wiring, outputs) do
    outputs
    |> Enum.map(&(&1 |> to_segments(wiring) |> number()))
    |> Enum.reverse()
    |> Enum.with_index()
    |> Enum.map(fn {x, n} -> x * trunc(:math.pow(10, n)) end)
    |> Enum.sum()
  end

  defp permutations([x]) do
    [x]
  end

  defp permutations(list) do
    for head <- list, tail <- permutations(list -- [head]) do
      [head | List.wrap(tail)]
    end
  end

  defp to_segments(wires, wiring) do
    Enum.map(wiring, fn x ->
      if x in wires do
        1
      else
        0
      end
    end)
  end

  defp number([1, 1, 1, 0, 1, 1, 1]), do: 0
  defp number([0, 0, 1, 0, 0, 1, 0]), do: 1
  defp number([1, 0, 1, 1, 1, 0, 1]), do: 2
  defp number([1, 0, 1, 1, 0, 1, 1]), do: 3
  defp number([0, 1, 1, 1, 0, 1, 0]), do: 4
  defp number([1, 1, 0, 1, 0, 1, 1]), do: 5
  defp number([1, 1, 0, 1, 1, 1, 1]), do: 6
  defp number([1, 0, 1, 0, 0, 1, 0]), do: 7
  defp number([1, 1, 1, 1, 1, 1, 1]), do: 8
  defp number([1, 1, 1, 1, 0, 1, 1]), do: 9
  defp number(_), do: :error
end

input
|> Kino.Input.read()
|> Part2.run()

Test

ExUnit.start(autorun: false)

defmodule Day8Tests do
  use ExUnit.Case, async: true

  setup do
    input = """
    be cfbegad cbdgef fgaecd cgeb fdcge agebfd fecdb fabcd edb | fdgacbe cefdb cefbgd gcbe
    edbfga begcd cbg gc gcadebf fbgde acbgfd abcde gfcbed gfec | fcgedb cgb dgebacf gc
    fgaebd cg bdaec gdafb agbcfd gdcbef bgcad gfac gcb cdgabef | cg cg fdcagb cbg
    fbegcd cbd adcefb dageb afcb bc aefdc ecdab fgdeca fcdbega | efabcd cedba gadfec cb
    aecbfdg fbg gf bafeg dbefa fcge gcbea fcaegb dgceab fcbdga | gecf egdcabf bgf bfgea
    fgeab ca afcebg bdacfeg cfaedg gcfdb baec bfadeg bafgc acf | gebdcfa ecba ca fadegcb
    dbcfg fgd bdegcaf fgec aegbdf ecdfab fbedc dacgb gdcebf gf | cefg dcbef fcge gbcadfe
    bdfegc cbegaf gecbf dfcage bdacg ed bedf ced adcbefg gebcd | ed bcgafe cdgba cbgef
    egadfb cdbfeg cegd fecab cgb gbdefca cg fgcdab egfdb bfceg | gbdfcae bgc cg cgb
    gcafb gcf dcaebfg ecagb gf abcdeg gaef cafbge fdbac fegbdc | fgae cfgab fg bagce
    """

    [input: input]
  end

  test "part1", %{input: input} do
    assert Part1.run(input) == 26
  end

  test "part2", %{input: input} do
    assert Part2.run(input) == 61_229
  end
end

ExUnit.run()