traversal.exs

defmodule LevelOrderTraversal do
  # Nós da árvore representados como {valor, {filho_direita, filho_esquerda}}
  @type tree :: {term, {tree | nil, tree | nil}}

  # Função que converte a árvore em seus níveis
  @spec traverse(tree) :: [[term]]
  def traverse(nil), do: [[]]

  def traverse({x, {l, r}}) do
    [[x] | Enum.zip_with(traverse(l), traverse(r), &Enum.concat/2)]
  end

  def traverse(x), do: [[x]]

  @spec print(tree) :: :ok
  def print(tree) do
    tree
    |> traverse()
    |> Enum.map(fn level ->
      level
      |> Enum.join(" ")
      |> IO.puts()
    end)

    :ok
  end
end