Skip to content

Commit

Permalink
Add binary-search-tree exercise (#161)
Browse files Browse the repository at this point in the history
  • Loading branch information
ageron authored Oct 24, 2024
1 parent 8b00510 commit e50f3e4
Show file tree
Hide file tree
Showing 8 changed files with 355 additions and 0 deletions.
8 changes: 8 additions & 0 deletions config.json
Original file line number Diff line number Diff line change
Expand Up @@ -571,6 +571,14 @@
"prerequisites": [],
"difficulty": 4
},
{
"slug": "binary-search-tree",
"name": "Binary Search Tree",
"uuid": "c0c96f3d-833e-4d15-b4ae-ccc711a0e165",
"practices": [],
"prerequisites": [],
"difficulty": 4
},
{
"slug": "custom-set",
"name": "Custom Set",
Expand Down
47 changes: 47 additions & 0 deletions exercises/practice/binary-search-tree/.docs/instructions.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
# Instructions

Insert and search for numbers in a binary tree.

When we need to represent sorted data, an array does not make a good data structure.

Say we have the array `[1, 3, 4, 5]`, and we add 2 to it so it becomes `[1, 3, 4, 5, 2]`.
Now we must sort the entire array again!
We can improve on this by realizing that we only need to make space for the new item `[1, nil, 3, 4, 5]`, and then adding the item in the space we added.
But this still requires us to shift many elements down by one.

Binary Search Trees, however, can operate on sorted data much more efficiently.

A binary search tree consists of a series of connected nodes.
Each node contains a piece of data (e.g. the number 3), a variable named `left`, and a variable named `right`.
The `left` and `right` variables point at `nil`, or other nodes.
Since these other nodes in turn have other nodes beneath them, we say that the left and right variables are pointing at subtrees.
All data in the left subtree is less than or equal to the current node's data, and all data in the right subtree is greater than the current node's data.

For example, if we had a node containing the data 4, and we added the data 2, our tree would look like this:

4
/
2

If we then added 6, it would look like this:

4
/ \
2 6

If we then added 3, it would look like this

4
/ \
2 6
\
3

And if we then added 1, 5, and 7, it would look like this

4
/ \
/ \
2 6
/ \ / \
1 3 5 7
24 changes: 24 additions & 0 deletions exercises/practice/binary-search-tree/.meta/Example.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
module [fromList, toList]

BinaryTree : [Nil, Node { value : U64, left : BinaryTree, right : BinaryTree }]

fromList : List U64 -> BinaryTree
fromList = \data ->
data |> List.walk Nil insert

insert : BinaryTree, U64 -> BinaryTree
insert = \tree, value ->
when tree is
Nil -> Node { value, left: Nil, right: Nil }
Node node ->
if value <= node.value then
Node { node & left: (node.left |> insert value) }
else
Node { node & right: (node.right |> insert value) }

toList : BinaryTree -> List U64
toList = \tree ->
when tree is
Nil -> []
Node node ->
node.left |> toList |> List.append node.value |> List.concat (node.right |> toList)
18 changes: 18 additions & 0 deletions exercises/practice/binary-search-tree/.meta/config.json
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
{
"authors": [
"ageron"
],
"files": {
"solution": [
"BinarySearchTree.roc"
],
"test": [
"binary-search-tree-test.roc"
],
"example": [
".meta/Example.roc"
]
},
"blurb": "Insert and search for numbers in a binary tree.",
"source": "Josh Cheek"
}
50 changes: 50 additions & 0 deletions exercises/practice/binary-search-tree/.meta/template.j2
Original file line number Diff line number Diff line change
@@ -0,0 +1,50 @@
{%- import "generator_macros.j2" as macros with context -%}
{{ macros.canonical_ref() }}
{{ macros.header() }}

import {{ exercise | to_pascal }} exposing [fromList, toList]

{% macro to_int_list(list) -%}
[{% for item in list %}{{ item }},{% endfor %}]
{%- endmacro %}

{% macro to_tree(tree) -%}
{% if tree is none %}Nil{% else %}Node {
value: {{ tree["data"] }},
left: {{ to_tree(tree["left"]) }},
right: {{ to_tree(tree["right"]) }},
}{% endif %}
{%- endmacro %}

{% for supercase in cases %}
##
## {{ supercase["description"] }}
##
{%- if supercase["cases"] %}
{%- set subcases = supercase["cases"] %}
{%- else %}
{%- set subcases = [supercase] %}
{%- endif %}

{% for subcase in subcases -%}
{%- if subcase["description"] != supercase["description"] %}
# {{ subcase["description"] }}
{%- endif %}
expect
data = {{ to_int_list(subcase["input"]["treeData"]) }}
{%- if subcase["property"] == "data" %}
result = data |> fromList
expected = {{ to_tree(subcase["expected"]) }}
result == expected
{%- else %}
tree = data |> fromList
result = tree |> toList
expected = {{ to_int_list(subcase["expected"]) }}
result == expected
{%- endif %}

{% endfor %}

{% endfor %}


40 changes: 40 additions & 0 deletions exercises/practice/binary-search-tree/.meta/tests.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
# This is an auto-generated file.
#
# Regenerating this file via `configlet sync` will:
# - Recreate every `description` key/value pair
# - Recreate every `reimplements` key/value pair, where they exist in problem-specifications
# - Remove any `include = true` key/value pair (an omitted `include` key implies inclusion)
# - Preserve any other key/value pair
#
# As user-added comments (using the # character) will be removed when this file
# is regenerated, comments can be added via a `comment` key.

[e9c93a78-c536-4750-a336-94583d23fafa]
description = "data is retained"

[7a95c9e8-69f6-476a-b0c4-4170cb3f7c91]
description = "insert data at proper node -> smaller number at left node"

[22b89499-9805-4703-a159-1a6e434c1585]
description = "insert data at proper node -> same number at left node"

[2e85fdde-77b1-41ed-b6ac-26ce6b663e34]
description = "insert data at proper node -> greater number at right node"

[dd898658-40ab-41d0-965e-7f145bf66e0b]
description = "can create complex tree"

[9e0c06ef-aeca-4202-b8e4-97f1ed057d56]
description = "can sort data -> can sort single number"

[425e6d07-fceb-4681-a4f4-e46920e380bb]
description = "can sort data -> can sort if second number is smaller than first"

[bd7532cc-6988-4259-bac8-1d50140079ab]
description = "can sort data -> can sort if second number is same as first"

[b6d1b3a5-9d79-44fd-9013-c83ca92ddd36]
description = "can sort data -> can sort if second number is greater than first"

[d00ec9bd-1288-4171-b968-d44d0808c1c8]
description = "can sort data -> can sort complex tree"
11 changes: 11 additions & 0 deletions exercises/practice/binary-search-tree/BinarySearchTree.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,11 @@
module [fromList, toList]

BinaryTree : [Nil, Node { value : U64, left : BinaryTree, right : BinaryTree }]

fromList : List U64 -> BinaryTree
fromList = \data ->
crash "Please implement the 'fromList' function"

toList : BinaryTree -> List U64
toList = \tree ->
crash "Please implement the 'toList' function"
157 changes: 157 additions & 0 deletions exercises/practice/binary-search-tree/binary-search-tree-test.roc
Original file line number Diff line number Diff line change
@@ -0,0 +1,157 @@
# These tests are auto-generated with test data from:
# https://github.com/exercism/problem-specifications/tree/main/exercises/binary-search-tree/canonical-data.json
# File last updated on 2024-10-21
app [main] {
pf: platform "https://github.com/roc-lang/basic-cli/releases/download/0.15.0/SlwdbJ-3GR7uBWQo6zlmYWNYOxnvo8r6YABXD-45UOw.tar.br",
}

main =
Task.ok {}

import BinarySearchTree exposing [fromList, toList]

##
## data is retained
##

expect
data = [4]
result = data |> fromList
expected = Node {
value: 4,
left: Nil,
right: Nil,
}
result == expected

##
## insert data at proper node
##

# smaller number at left node
expect
data = [4, 2]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 2,
left: Nil,
right: Nil,
},
right: Nil,
}
result == expected

# same number at left node
expect
data = [4, 4]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 4,
left: Nil,
right: Nil,
},
right: Nil,
}
result == expected

# greater number at right node
expect
data = [4, 5]
result = data |> fromList
expected = Node {
value: 4,
left: Nil,
right: Node {
value: 5,
left: Nil,
right: Nil,
},
}
result == expected

##
## can create complex tree
##

expect
data = [4, 2, 6, 1, 3, 5, 7]
result = data |> fromList
expected = Node {
value: 4,
left: Node {
value: 2,
left: Node {
value: 1,
left: Nil,
right: Nil,
},
right: Node {
value: 3,
left: Nil,
right: Nil,
},
},
right: Node {
value: 6,
left: Node {
value: 5,
left: Nil,
right: Nil,
},
right: Node {
value: 7,
left: Nil,
right: Nil,
},
},
}
result == expected

##
## can sort data
##

# can sort single number
expect
data = [2]
tree = data |> fromList
result = tree |> toList
expected = [2]
result == expected

# can sort if second number is smaller than first
expect
data = [2, 1]
tree = data |> fromList
result = tree |> toList
expected = [1, 2]
result == expected

# can sort if second number is same as first
expect
data = [2, 2]
tree = data |> fromList
result = tree |> toList
expected = [2, 2]
result == expected

# can sort if second number is greater than first
expect
data = [2, 3]
tree = data |> fromList
result = tree |> toList
expected = [2, 3]
result == expected

# can sort complex tree
expect
data = [2, 1, 3, 6, 7, 5]
tree = data |> fromList
result = tree |> toList
expected = [1, 2, 3, 5, 6, 7]
result == expected

0 comments on commit e50f3e4

Please sign in to comment.