chandan374

chandan374

Build binary tree from level order array

I am new to elixer, I need some help to construct binary tree from list of element(given in level order). Due to immutability i’m facing difficutly to implement this.
For ex: [1, 2, 3, -1, -1, -1, -1] will be represented as:

                  1
                /   \
              2       3

Here -1 means, node will be null.

Expected Output Format:

TreeNode{
	data: 1,
	left: TreeNode{
		data: 2,
		left: nil,
		right: nil
	},
	right: TreeNode{
		data: 3,
		left: nil,
		right: nil
	}	
}

In ruby, we can implement in this way.

def input_tree(input)
	val = input[0]
	input.shift()
	root = TreeNode.new(val)
	queue = []
	queue.push(root)
	
	while queue.size > 0
		current_node = queue[0]
		queue.shift()
		
		leftValue = input[0]
		input.shift()
	
		rightValue = input[0]
		input.shift()
		
		if leftValue != -1
			leftNode = TreeNode.new(leftValue)
			current_node.left = leftNode
			queue.push(leftNode)
		end
		
		if rightValue != -1
			rightNode = TreeNode.new(rightValue)
			current_node.right = rightNode
			queue.push(rightNode)
		end
	end
	return root
end

TIA :slight_smile:

Marked As Solved

al2o3cr

al2o3cr

Here’s an approach that uses a pair of recursive functions:

defmodule TreeNode do
  defstruct ~w[data left right]a
end

defmodule TreeBuild do
  @nothing -1

  def run([root | rest]) do
    [[left, right]] = subnodes(1, rest)
    %TreeNode{data: root, left: left, right: right}
  end

  def subnodes(0, []), do: []
  def subnodes(count, input) do
    {roots, rest} = Enum.split(input, 2*count)

    next_count = Enum.count(roots, & &1 != @nothing)

    child_nodes = subnodes(next_count, rest)

    roots
    |> build_nodes(child_nodes, [])
    |> Enum.chunk_every(2, 2, [nil])
  end

  def build_nodes([], _, acc), do: Enum.reverse(acc)
  def build_nodes([@nothing | roots], child_nodes, acc) do
    build_nodes(roots, child_nodes, [nil | acc])
  end
  def build_nodes([root | roots], [], acc) do
    node = %TreeNode{data: root, left: nil, right: nil}
    build_nodes(roots, [], [node | acc])
  end
  def build_nodes([root | roots], [[left, right] | child_nodes], acc) do
    node = %TreeNode{data: root, left: left, right: right}
    build_nodes(roots, child_nodes, [node | acc])
  end
end

TreeBuild.run([1, 2, 3, -1, -1, -1, -1])

TreeBuild.run([5,4,8,11,-1,17,4,7,-1,-1,-1,5, -1, -1, -1, -1])

Some notes on the implementation:

  • the sample input has one fewer -1 on the end than I expected at first; this version of build_nodes is tolerant of any number of trailing -1s.
  • chunk_every is used to tidily handle cases where build_nodes returns an odd number of nodes by providing a nil for leftovers

@Eiji I believe the error in your diagram is the two -1s under the -1 that’s a right-child of 4. -1s on a given level shouldn’t consume any input values in the next level.

Also Liked

Eiji

Eiji

(…) if a node has an index i , its children are found at indices 2i+1 (for the left child) and 2i+2 (for the right) (…)

Source: Arrays at Binary tree | Wikipedia

With above writing code is really simple:

defmodule TreeNode do
  defstruct ~w[data left right]a
end

defmodule Example do
  def sample(input, index \\ 0) do
    data = Enum.at(input, index)

    unless data in [nil, -1] do
      %TreeNode{
        data: data,
        left: sample(input, index * 2 + 1),
        right: sample(input, index * 2 + 2)
      }
    end
  end
end

iex> Example.sample([1, 2, 3, -1, -1, -1, -1])
%TreeNode{
  data: 1,
  left: %TreeNode{data: 2, left: nil, right: nil},
  right: %TreeNode{data: 3, left: nil, right: nil}
}
cevado

cevado

have you tried using gb_trees from erlang?
https://www.erlang.org/doc/man/gb_trees.html

benwilson512

benwilson512

Author of Craft GraphQL APIs in Elixir with Absinthe

Hi @chandan374 can you show what you’ve tried so far?

Eiji

Eiji

The tree representation you provided is incorrect as 5 leaf should be a left child of 17 node. Therefore I think my example is working.

This is how it should look:

                           5
                      /         \
                     4            8
                   /   \        /    \
                  11    -1     17     4
                 /  \   / \   / \    / \
                7   -1 -1 -1  5 -1  -1 -1
               /
             -1

# after removing -1 entries

                        5
                      /   \
                     4     8
                    /     /  \
                   11    17   4
                  /     /
                 7     5

To prove that I have also wrote another solution. This time my code is building the tree in reverse mode (I’m starting with leafs) based on pattern matching.

defmodule TreeNode do
  defstruct ~w[data left right]a
end

defmodule Example do
  def sample(input) do
    input |> split() |> transform(nil)
  end

  # we split a list by count
  # on top is only root, so default count is always 1
  # every other level have count * 2 nodes
  defp split(list, acc \\ [], count \\ 1) do
    {left, right} = Enum.split(list, count)

    # we call split until right side is empty
    if right == [] do
      [left | acc]
    else
      # after each split the left side is added to acc
      # therefore we would have leafs list as fist item in acc
      # and a single root data as last item of acc
      split(right, [left | acc], count * 2)
    end
  end

  defp transform([leafs | groups], nil) do
    # transform leafs into initial acc
    # which would be used in transform groups 
    transform(groups, {Enum.map(leafs, &transform_leaf/1), []})
  end

  # finishing by returning root node
  defp transform([], {[root_node], []}), do: root_node

  defp transform([[] | groups], {[], acc_right}) do
    # reached end of group
    # all nodes we have collected in acc_right
    # are moved to acc_left
    # so they would be used in next group
    transform(groups, {Enum.reverse(acc_right), []})
  end

  defp transform([[-1 | group] | groups], {[], acc_right}) do
    # skipping -1 with no fake leafs
    transform([group | groups], {[], [nil | acc_right]})
  end

  defp transform([[-1 | group] | groups], {[nil, nil | acc_left], acc_right}) do
    # skipping -1 with transformed fake leafs
    transform([group | groups], {acc_left, [nil | acc_right]})
  end

  defp transform([[data | group] | groups], {[], acc_right}) do
    # node #{data} does not have children
    transform([group | groups], {[], [transform_node(data) | acc_right]})
  end

  defp transform([[data | group] | groups], {[left], acc_right}) do
    # node #{data} have only left child
    transform([group | groups], {[], [transform_node(data, left) | acc_right]})
  end

  defp transform([[data | group] | groups], {[left, right | acc_left], acc_right}) do
    # node #{data} have both children
    transform([group | groups], {acc_left, [transform_node(data, left, right) | acc_right]})
  end

  defp transform_leaf(-1), do: nil
  defp transform_leaf(data), do: %TreeNode{data: data}

  defp transform_node(data, left \\ nil, right \\ nil) do
    # transforming node
    %TreeNode{data: data, left: left, right: right}
  end
end

iex> Example.sample([5, 4, 8, 11, -1, 17, 4, 7, -1, -1, -1, 5, -1, -1, -1, -1])
%TreeNode{
  data: 5,
  left: %TreeNode{
    data: 4,
    left: %TreeNode{
      data: 11,
      left: %TreeNode{data: 7, left: nil, right: nil},
      right: nil
    },
    right: nil
  },
  right: %TreeNode{
    data: 8,
    left: %TreeNode{
      data: 17,
      left: %TreeNode{data: 5, left: nil, right: nil},
      right: nil
    },
    right: %TreeNode{data: 4, left: nil, right: nil}
  }
}

Both examples returns exactly same data. Feel free to ask if you have any questions. :smiling_imp:

Where Next?

Popular in Questions Top

Brian
What is the proper way to load a module from a file in to IEX? In the python world, doing something like this pretty standard: from ....
New
LegitStack
I’m hoping you guys can give me some general advice and perhaps code examples if you’re feeling up to it. I’m very interested in Elixir,...
New
New
sacepums
Hey guys. I'm new to elixir and im really stocked about it. But I ran into a bit of problem - I need to convert a date sting, for examp...
New
lk-geimfari
What is most correct way to open, read and parse JSON file with poison? For example if we have example.json file in root of some projec...
New
Phillipp
Hey, I have a NanoPi-M3 and try to install Elixir on their Ubuntu image. I followed the Raspberry Pi installation instructions from the ...
New
qwerescape
Is there a way to get the call stack or stack trace at any point in the code? Not from exceptions, but an expression that returns how the...
New
chewm
Hi guys, nice to meet you to the whole forum, I’m new here, I’m trying to configure visual studio code for elixir, right now the intellis...
New
jay1
Why is it that the mnesia database isn’t the most preferred database for use in Elixir/Phoenix?
New
joeerl
Hello again - after a longish gap I’ve decided I really must dig into Elixir and see what’s been happening here - so I have a few questio...
New

Other popular topics Top

JakeBecker
TL;DR: I’ve just released an implementation of Microsoft’s IDE-independent Language Server Protocol for Elixir. It adds language support ...
1140 51847 244
New
vertexbuffer
Hello, can anybody help here..? I have a list of players and I what to delete an element, but every for loop the list is reverting to ori...
New
lastday4you
I wanted to check elixir version in phoenix because i found that my elixir is 1.5 but when i use Enum.chunk_by it said the function is un...
New
shahryarjb
Hello, I have map which I want to convert it to string like this: the map: %{last_name: "tavakkoli", name: "shahryar"} the string I ne...
New
ovidiubadita
Hey all, I discovered Elixir and I love it. I always wanted to learn a functional programming and I intended to go for Haskell, but afte...
New
baxterw3b
Hi guys, i’m new in the Elixir world, and i have to say, that i love it! i’m having some problem to understand anonymous functions with ...
New
chrisalley
ExUnit now has describe blocks which is a welcome addition coming from RSpec. In the docs, it states that nested hierarchies of describe ...
New
9mm
I am constructing a JSON object (map) and I need to conditionally set a field. I’m trying to write proper elixir-way code… and I’m at a l...
New
aesmail
Hello guys, I have finally made it. I created an admin interface for a framework. It’s been on my todo list for years and with the curre...
New
vrod
I am using the Starship cross-shell prompt – it seems pretty nice, but I get some errors: [WARN] - (starship::utils): Executing command ...
New

We're in Beta

About us Mission Statement